From 47b9376ac0f7c798fb95759e950f994923b22271 Mon Sep 17 00:00:00 2001 From: atenagm1375 Date: Thu, 25 Mar 2021 00:06:51 +0430 Subject: [PATCH] fix allocation problem --- cnsproject/network/neural_populations.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cnsproject/network/neural_populations.py b/cnsproject/network/neural_populations.py index 7356a4b..9ec48e5 100644 --- a/cnsproject/network/neural_populations.py +++ b/cnsproject/network/neural_populations.py @@ -57,7 +57,7 @@ def __init__( self.additive_spike_trace = additive_spike_trace if self.spike_trace: - self.register_buffer("traces", torch.Tensor()) + self.register_buffer("traces", torch.zeros(*self.shape)) self.register_buffer("tau_s", torch.tensor(tau_s)) if self.additive_spike_trace: @@ -68,7 +68,7 @@ def __init__( self.is_inhibitory = is_inhibitory self.learning = learning - self.register_buffer("s", torch.ByteTensor()) + self.register_buffer("s", torch.zeros(*self.shape, dtype=torch.bool)) self.dt = None @abstractmethod