chore: Update refractory period calculation in IFNodes_PredInf

BindsNET · Jul 22, 2024 · 883f29b · 883f29b
1 parent 5dc62b8
commit 883f29b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindsnet/network/nodes.py b/bindsnet/network/nodes.py
@@ -1766,7 +1766,8 @@ def __init__(
 
         self.Fire_only_Once = Fire_only_Once
         if self.Fire_only_Once:
-            self.refrac = torch.iinfo(torch.int32).max
+            self.refrac = torch.ones_like(self.refrac) * torch.iinfo(torch.int32).max
+
 
         self.lbound = lbound  # Lower bound of voltage.