summaryrefslogtreecommitdiff
path: root/neurobench_testing/examples/gsc/SNN.py
diff options
context:
space:
mode:
authorTanner Robison <[email protected]>2026-07-06 10:35:32 -0700
committerTanner Robison <[email protected]>2026-07-07 15:42:48 -0700
commite2b30976cdaccf9d2b9820fefa22ced03c82711f (patch)
tree99f1d6d5824882a533f46f1819138ec5196abed3 /neurobench_testing/examples/gsc/SNN.py
parent3a17db637fd41015df26f1cceb0d1c098e85d436 (diff)
readout layer trained
verified with time-series forecasting task and plotted
Diffstat (limited to 'neurobench_testing/examples/gsc/SNN.py')
-rw-r--r--neurobench_testing/examples/gsc/SNN.py17
1 files changed, 0 insertions, 17 deletions
diff --git a/neurobench_testing/examples/gsc/SNN.py b/neurobench_testing/examples/gsc/SNN.py
deleted file mode 100644
index 1642e28..0000000
--- a/neurobench_testing/examples/gsc/SNN.py
+++ /dev/null
@@ -1,17 +0,0 @@
-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