Transfer learning for computer vision example.

Convolutional neural network for classifying ant and bee images.

To get requirements, run the following commands (note this assumes your GPU is CUDA-enabled):

• pip install -r requirements.txt
• pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

Download the dataset from this link

To run the program, run the following command:

• python main.py