Skip to content

ThuryW/ResNet20-for-Cifar10

Repository files navigation

ResNet20 for Cifar10

How to run

Just run

python train.py # --lr 0.001 ...
python test.py  # --path ./checkpoint/yourfilename

You can add args if you wanna change hyperparameters, see train.py.

Use -r to resume from checkpoint.

Export weights

Save the model weights in csv files

python export_weights.py

Train result

Network Top-1 accuracy
ResNet20 92.23%
ResNet32 TBD
ResNet44 TBD

About

ResNet20 implement for Cifar10 dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published