The files train_mnist.py
, train_cifar10.py
and train_imagenet.py
are the main scripts to train on MNIST, CIFAR-10 and ImageNet, respectively.
cswin / burstprop Goto Github PK
View Code? Open in Web Editor NEWCode to train on MNIST, CIFAR-10 and ImageNet using burstprop.
Home Page: https://www.biorxiv.org/content/10.1101/2020.03.30.015511v1