BINDS-LAB-UMASS / bindsnet_experiments

Collection of experimental Python and accompanying Bash scripts for training and evaluation of BindsNET networks.
66 stars 12 forks source link

Make fully_connected_mnist.py work with BindsNET 0.3.1 #9

Open 0xnurl opened 1 year ago

0xnurl commented 1 year ago

fully_connected_mnist.py is a demo script for using the conversion module in BindsNET to convert between a multilayer PyTorch net to a SNN. Some interfaces changed in BindsNET which prevented the script from working. This commit updates the script and the SNN converted from a trained ANN reaches ~95%+ accuracy on the MNIST training set.