davrot/pytorch-sbs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
David Rotermund 2023-02-02 19:23:38 +01:00 committed by GitHub
parent d51738832c
commit 3b70ae789a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
network/SbS.py
View file

@ -130,9 +130,9 @@ class SbS(torch.nn.Module):
global_spike_size.append(torch.tensor([0, 0, 0, 0])) global_spike_size.append(torch.tensor([0, 0, 0, 0]))
if device == torch.device("cpu"): if device == torch.device("cpu"):
global_sbs_hdynamic_cpp.append(HDynamicCNNGPU())
else:
global_sbs_hdynamic_cpp.append(HDynamicCNNCPU()) global_sbs_hdynamic_cpp.append(HDynamicCNNCPU())
else:
global_sbs_hdynamic_cpp.append(HDynamicCNNGPU())
if device == torch.device("cpu"): if device == torch.device("cpu"):
global_spike_generation_cpp.append(SpikeGenerationCPU()) global_spike_generation_cpp.append(SpikeGenerationCPU())