Model Card: CCT Model for Organism Classification

Model Overview

Training Details

• Dataset: A custom dataset containing images from the animal and plant classes of the ImageNet1k dataset.
• Task: Organism classification, where the model categorizes images into specific organism classes.
• Model Architecture: Convolutional Contrastive Transformer (CCT).
• Training Parameters: The model was trained using the parameters defined in the args.yml file that was uploaded.

Available Checkpoints

1. model_best_sd.pth:
   • This checkpoint can be directly used for classification tasks.
2. model_best.pth.tar:
   • This checkpoint contains parameters from previous training iterations and can be used for further fine-tuning or continued training.

How to Use

1. For Classification: Use the pre-trained model available in model_best_sd.pth to directly perform image classification on the Organism classes.
2. For Fine-Tuning: If you wish to continue training the model with your own dataset or further fine-tune it, you can start from the checkpoint model_best.pth.tar. This checkpoint includes the parameters from the previous training and allows for continued training.

Usage Instructions