preact resnet with test on cifar10

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

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