Skip to content

Latest commit

 

History

History
22 lines (15 loc) · 562 Bytes

README.md

File metadata and controls

22 lines (15 loc) · 562 Bytes

Training CIFAR-10 with ResNet using Pytorch

This is the first group project of ECE7123 Deep Learning I took at NYU Tandon in Spring 2022.

MiniProject1_G26

Download the CIFAR-10 data from here: http://www.cs.toronto.edu/~kriz/cifar.html And put it in the ./data folder.

checkpoints

checkpoints folder contains our trained models and their corresponding training loss, testing loss and testing accuracy during training.

training

python training/main.py

loading trained model and testing

python training/evaluate_checkpoint.py