preact resnet with test on cifar10

metya 86d77512ff Update requirements.txt před 3 roky
.idea 08240a8067 Initial commit před 6 roky
models 22576ddbc9 před 6 roky
.gitignore 07413ecdad před 6 roky
README.md ee5d00b103 typo fix před 6 roky
main.py cdbdb022df před 6 roky
requirements.txt 86d77512ff Update requirements.txt před 3 roky

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