pytorch-sbs/README.md

41 lines
1.4 KiB
Markdown
Raw Normal View History

2022-04-30 02:03:34 +02:00
# pytorch-sbs
SbS Extension for PyTorch
2022-04-30 02:14:02 +02:00
2022-04-30 13:45:04 +02:00
# Based on these scientific papers
2022-04-30 02:14:02 +02:00
2022-04-30 02:16:43 +02:00
**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
2022-04-30 02:14:24 +02:00
2022-04-30 02:16:43 +02:00
**Efficient Computation Based on Stochastic Spikes**
Udo Ernst, David Rotermund, and Klaus Pawelzik
2022-04-30 02:17:18 +02:00
Neural Computation (2007) 19 (5): 13131343. https://doi.org/10.1162/neco.2007.19.5.1313
2022-04-30 02:16:43 +02:00
https://direct.mit.edu/neco/article-abstract/19/5/1313/7183/Efficient-Computation-Based-on-Stochastic-Spikes
2022-04-30 13:45:04 +02:00
# 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++
You need to modify the Makefile in the C++ directory to your Python installation.
2023-01-05 13:32:32 +01:00
2023-01-05 13:29:12 +01:00
In addition your Python installation needs the PyBind11 package installed. You might want to perform a
2022-04-30 13:45:04 +02:00
pip install pybind11
The Makefile uses clang as a compiler. If you want something else then you need to change the Makefile.
2023-01-05 13:29:12 +01:00
For CUDA I used version 12.0.
2023-01-15 16:13:55 +01:00
2023-01-15 16:14:26 +01:00
# Config files and pre-existing weights
2023-01-15 16:13:55 +01:00
Three .json config files are required:
dataset.json : Information about the dataset
network.json : Describes the network architecture
def.json : Controlls the other parameters
If you want to load existing weights, just put them in a sub-folder called Previous