connections

Author: n-shevko

bindsnet/network/network.py

@@ -433,29 +433,30 @@ def run(
             torch.cuda.synchronize()
 
             for c in self.connections:
-                flad_m = False
-                if A_Minus != None and ((isinstance(A_Minus, float)) or (c in A_Minus)):
-                    if A_MD:
-                        kwargs["a_minus"] = A_Minus[c]
-                    else:
-                        kwargs["a_minus"] = A_Minus
-                    flad_m = True
+                with stream():
+                    flad_m = False
+                    if A_Minus != None and ((isinstance(A_Minus, float)) or (c in A_Minus)):
+                        if A_MD:
+                            kwargs["a_minus"] = A_Minus[c]
+                        else:
+                            kwargs["a_minus"] = A_Minus
+                        flad_m = True
 
-                flad_p = False
-                if A_Plus != None and ((isinstance(A_Plus, float)) or (c in A_Plus)):
-                    if A_PD:
-                        kwargs["a_plus"] = A_Plus[c]
-                    else:
-                        kwargs["a_plus"] = A_Plus
-                    flad_p = True
-
-                self.connections[c].update(
-                    mask=masks.get(c, None), learning=self.learning, **kwargs
-                )
-                if flad_m:
-                    kwargs.pop("a_minus")
-                if flad_p:
-                    kwargs.pop("a_plus")
+                    flad_p = False
+                    if A_Plus != None and ((isinstance(A_Plus, float)) or (c in A_Plus)):
+                        if A_PD:
+                            kwargs["a_plus"] = A_Plus[c]
+                        else:
+                            kwargs["a_plus"] = A_Plus
+                        flad_p = True
+
+                    self.connections[c].update(
+                        mask=masks.get(c, None), learning=self.learning, **kwargs
+                    )
+                    if flad_m:
+                        kwargs.pop("a_minus")
+                    if flad_p:
+                        kwargs.pop("a_plus")
 
             # # Get input to all layers.
             # current_inputs.update(self._get_inputs())