The repo contains the code associated with the SNN training method ASF-BP. The code has been tested with Pytorch 1.1.0 and Python 3.7.4.
To train a new model from scratch, the basic syntax is like: python vgg7.py
To test a pre-trained model, the basic syntax is likepython vgg7.py --resume model_bestT1_cifar10_v7.pth.tar --evaluate
Chankyu Lee, Syed Shakib Sarwar, Priyadarshini Panda, Gopalakrishnan Srinivasan, and Kaushik Roy. Enabling spike-based backpropagation for training deep neural network architectures. Frontiers in Neuroscience, 14, 2020.