The implementation of S4NN presented in "S. R. Kheradpisheh and T. Masquelier, Temporal backpropagation for spiking neural networks with one spike per neuron, International Journal of Neural Systems (2020), doi: 10.1142/S0129065720500276", availbale at: https://www.worldscientific.com/doi/10.1142/S0129065720500276. it is also available on arXiv at https://arxiv.org/abs/1910.09495.
Two versions of the code are available:
To run the codes on MNIST dataset, you should first unzip the MNIST.zip file. Then, you should install the python-mnist package. To do so, you can run the following command:
$ sudo pip install python-mnist
If you want to run the codes on GPU, you should set GPU=True. Also, you need to install Cupy package to work with GPU. Cupy is already installed on Google CoLab. You can install it on your own machine by the following command:
$ sudo pip install cupy
The pre-trained wight matrix is available at weights_pretrained.npy file.