sunzhe839 / SNU_Chainer

Spiking Neural Unit (SNU) implemented with Chainer

SNU Chainer

Spiking neural unit implemented with Chainer-v5
See: https://arxiv.org/abs/1812.07040

Requirements

• Chainer v5
• cupy
• tqdm

Usage

Train the network

Run, python train.py -g 0 -b 128 -e 100 -t 10
(gpu, batch, epoch, simulation time steps)

Test output activity

Run, python analysis.py -g 0 -m ./results/model

Check SNU layer

Run, python check_snu_layer.py

Check Jittered MNIST

Run, check_jittered_mnist.py