Init

Author: n-shevko

bindsnet/utils.py

@@ -276,3 +276,18 @@ def reshape_local_connection_2d_weights(
                             ]
 
         return square
+
+
+class FakeStream:
+    def __enter__(self):
+        pass
+
+    def __exit__(self, *_):
+        pass
+
+
+def stream(device=None):
+    try:
+        return torch.cuda.stream(torch.cuda.Stream(device=device))
+    except RuntimeError as _:
+        return FakeStream()

mutability.py

@@ -0,0 +1,19 @@
+import torch
+from torch.multiprocessing.spawn import spawn
+
+
+def layer_update(z, self):
+    self.x += 1
+
+
+class Network:
+    def __init__(self):
+        self.x = torch.tensor([1, 1, 1], device='cpu')
+
+    def run(self):
+        spawn(layer_update, args=(self,))
+        print(self.x)
+
+if __name__ == '__main__':
+    Network().run()
+