preact resnet with test on cifar10

metya cdbdb022df 6 лет назад
.idea 08240a8067 Initial commit 6 лет назад
models 22576ddbc9 6 лет назад
.gitignore 07413ecdad 6 лет назад
README.md cdbdb022df 6 лет назад
main.py cdbdb022df 6 лет назад
requirements.txt 22576ddbc9 6 лет назад

README.md

PreAct ResNet with some tests on cifar10

To run train without tests just do

python main.py --train --epochs num

To run train alongside with some unit tests just do

python main.py --test --train --epochs num

If you want to run just tests, then do

python main.py --test

###TODO

  • Regression metrics tests
  • Gradient explosion and gradient vanishing check
  • Pythonistic Unit Tests