GitHub - ThuryW/ResNet-for-Cifar10: ResNet20, ResNet32 and RestNet44 for Cifar10 dataset, train and export weight file

ResNet20 for Cifar10

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.

Save the model weights in csv files

python export_weights.py

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
model		model
.gitignore		.gitignore
README.md		README.md
export_weights.py		export_weights.py
hook.py		hook.py
test.py		test.py
train.py		train.py
utils.py		utils.py