Variable Size Image Classifier

This project implements a variable-size image classifier using Vision Transformers (ViT) and attention mechanisms.

Setup

1. Clone the repository:

   git clone https://github.com/your-username/variable-size-image-classifier.git
   cd variable-size-image-classifier
   

2. Create a virtual environment and activate it:

   python -m venv venv
   source venv/bin/activate  # On Windows, use `venv\Scripts\activate`
   

3. Install the required packages:

   pip install -r requirements.txt
   

4. Set up Weights & Biases:

   wandb login
   

Usage

Training

To train the model: