From 3b87b8bd271ade1622149516ae14f94b7ff20dd5 Mon Sep 17 00:00:00 2001 From: Tanner Robison Date: Wed, 17 Jun 2026 18:14:20 -0700 Subject: neurobench & memtorch --- neurobench_testing/examples/gsc/SNN.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 neurobench_testing/examples/gsc/SNN.py (limited to 'neurobench_testing/examples/gsc/SNN.py') diff --git a/neurobench_testing/examples/gsc/SNN.py b/neurobench_testing/examples/gsc/SNN.py new file mode 100644 index 0000000..1642e28 --- /dev/null +++ b/neurobench_testing/examples/gsc/SNN.py @@ -0,0 +1,17 @@ +from torch import nn +import snntorch as snn +from snntorch import surrogate + +beta = 0.9 +spike_grad = surrogate.fast_sigmoid() +net = nn.Sequential( + nn.Flatten(), + nn.Linear(20, 256), + snn.Leaky(beta=beta, spike_grad=spike_grad, init_hidden=True), + nn.Linear(256, 256), + snn.Leaky(beta=beta, spike_grad=spike_grad, init_hidden=True), + nn.Linear(256, 256), + snn.Leaky(beta=beta, spike_grad=spike_grad, init_hidden=True), + nn.Linear(256, 35), + snn.Leaky(beta=beta, spike_grad=spike_grad, init_hidden=True, output=True), +) \ No newline at end of file -- cgit v1.2.3