diff options
Diffstat (limited to 'neurobench_testing/examples/gsc/SNN.py')
| -rw-r--r-- | neurobench_testing/examples/gsc/SNN.py | 17 |
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 |