preact resnet with test on cifar10
metya
86d77512ff
Update requirements.txt
|
há 3 anos atrás | |
|---|---|---|
| .idea | há 6 anos atrás | |
| models | há 6 anos atrás | |
| .gitignore | há 6 anos atrás | |
| README.md | há 6 anos atrás | |
| main.py | há 6 anos atrás | |
| requirements.txt | há 3 anos atrás |
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