search

NeuromorphicProcessorProject / snn_toolbox

Toolbox for converting analog to spiking neural networks (ANN to SNN), and running them in a spiking neuron simulator.
MIT License
360 stars 104 forks source link

How do I get the BRIAN2 model of the converted SNN #106

Closed nishraptor closed 2 years ago

nishraptor commented 3 years ago

I want to convert the example pytorch ANN to an SNN, and get the brian model (not to run it). I then want to extract the connectivity of the entire network. How would I go about doing this?

rbodo commented 3 years ago

I'd start by modifying this example script for pytorch to use the brian backend, as in this example. If the scripts work for you with MNIST, then insert your dataset and model.

If using Brian via PyNN, the connections are saved automatically as text files, see here.

With Brian2, check out this implementation for saving the weights.