preact resnet with test on cifar10

metya cdbdb022df hace 6 años
.idea 08240a8067 Initial commit hace 6 años
models 22576ddbc9 hace 6 años
.gitignore 07413ecdad hace 6 años
README.md cdbdb022df hace 6 años
main.py cdbdb022df hace 6 años
requirements.txt 22576ddbc9 hace 6 años

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