From 3b70ae789a3f25491470dff0eab1bb53734b5aba Mon Sep 17 00:00:00 2001 From: David Rotermund <54365609+davrot@users.noreply.github.com> Date: Thu, 2 Feb 2023 19:23:38 +0100 Subject: [PATCH] Add files via upload --- network/SbS.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/network/SbS.py b/network/SbS.py index 72b141d..7bf43ce 100644 --- a/network/SbS.py +++ b/network/SbS.py @@ -130,9 +130,9 @@ class SbS(torch.nn.Module): global_spike_size.append(torch.tensor([0, 0, 0, 0])) if device == torch.device("cpu"): - global_sbs_hdynamic_cpp.append(HDynamicCNNGPU()) - else: global_sbs_hdynamic_cpp.append(HDynamicCNNCPU()) + else: + global_sbs_hdynamic_cpp.append(HDynamicCNNGPU()) if device == torch.device("cpu"): global_spike_generation_cpp.append(SpikeGenerationCPU())