davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
SbS Extension for PyTorch
10 commits 1 branch 0 tags 31 MiB
Python 45.6%
Cuda 33.1%
C++ 16.6%
Makefile 4.4%
C 0.2%
Other 0.1%
Find a file
David Rotermund 7d2908c751
Update README.md
2022-04-30 13:45:04 +02:00
C++ C++ module 2022-04-30 02:07:55 +02:00
DATA_CIFAR10 Example data converter 2022-04-30 02:06:41 +02:00
DATA_FASHION_MNIST Example data converter 2022-04-30 02:06:41 +02:00
DATA_MNIST Example data converter 2022-04-30 02:06:41 +02:00
Dataset.py First version 2022-04-30 02:07:09 +02:00
learn_it.py LR Scheduler: Separated the eps and w parameters 2022-04-30 13:40:51 +02:00
LICENSE Initial commit 2022-04-30 02:03:34 +02:00
Parameter.py First version 2022-04-30 02:07:09 +02:00
PyHDynamicCNNManyIP.pyi First version 2022-04-30 02:07:09 +02:00
PySpikeGeneration2DManyIP.pyi First version 2022-04-30 02:07:09 +02:00
README.md Update README.md 2022-04-30 13:45:04 +02:00
SbS.py First version 2022-04-30 02:07:09 +02:00

README.md

pytorch-sbs

SbS Extension for PyTorch

Based on these scientific papers

Back-Propagation Learning in Deep Spike-By-Spike Networks
David Rotermund and Klaus R. Pawelzik
Front. Comput. Neurosci., https://doi.org/10.3389/fncom.2019.00055
https://www.frontiersin.org/articles/10.3389/fncom.2019.00055/full

Efficient Computation Based on Stochastic Spikes
Udo Ernst, David Rotermund, and Klaus Pawelzik
Neural Computation (2007) 19 (5): 1313–1343. https://doi.org/10.1162/neco.2007.19.5.1313
https://direct.mit.edu/neco/article-abstract/19/5/1313/7183/Efficient-Computation-Based-on-Stochastic-Spikes

Python

It was programmed with 3.10.4. And I used some 3.10 Python expression. Thus you might get problems with older Python versions.

C++

It works without compiling the C++ modules. However it is 10x slower.
You need to modify the Makefile in the C++ directory to your Python installation.
In addition yoir Python installation needs the PyBind11 package installed. You might want to perform a
pip install pybind11
The Makefile uses clang as a compiler. If you want something else then you need to change the Makefile.