IsurangaPerera / CIFAR-10

CSCE 636 Fall 2022 Deep Learning Project

Additional Models and Log Files

Additional checkpoints can be located through this Google Drive Folder:

Requirements:

• Python 3.8.6
• Nvidia driver >=450
• Pytorch 1.7
• tqdm

Install packages via via:

pip3 install tqdm numpy torch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 tqdm scipy sklearn matplotlib

Running

The best performing model is included in the saved_models folder to reproduce results

Training

cd code && python3 main.py train

Testing

cd code && python3 main.py test --checkpoint 'best_model.pth'

Predictions

cd code && python3 main.py predict --checkpoint 'best_model.pth' --save_dir '../'