fix trace_decay buffer

cnrl · Mar 24, 2021 · 2715651 · 2715651
1 parent 055429f
commit 2715651
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cnsproject/network/neural_populations.py b/cnsproject/network/neural_populations.py
@@ -63,7 +63,7 @@ def __init__(
             if self.additive_spike_trace:
                 self.register_buffer("trace_scale", torch.tensor(trace_scale))
 
-            self.register_buffer("trace_decay", torch.empty_like(tau_s))
+            self.register_buffer("trace_decay", torch.empty_like(self.tau_s))
 
         self.is_inhibitory = is_inhibitory
         self.learning = learning