fix

Author: n-shevko

bindsnet/network/topology.py

@@ -441,7 +441,7 @@ def compute(self, s: torch.Tensor) -> torch.Tensor:
 
         # Run through pipeline
         for f in self.pipeline:
-            if type(f).__name__ == 'Weight' and self.mask:
+            if type(f).__name__ == 'Weight' and self.mask is not None:
                 f.value.masked_fill_(self.mask, 0)
             conn_spikes = f.compute(conn_spikes)
 

examples/dosidicus/network.py

@@ -26,15 +26,13 @@
     source=source_layer,
     target=target_layer,
     pipeline=[weight, bias],
+    mask=~torch.tril(torch.ones((5, 5)), diagonal=-1).bool(),
     device='cpu'
 )
 network.add_connection(connection, source="input", target="output")
 print(connection.pipeline[0].value)
 network.run(
     inputs={"input": torch.bernoulli(torch.rand(250, 5)).byte()},
-    time=250,
-    masks={
-       ('input', 'output'): ~torch.tril(torch.ones((5, 5)), diagonal=-1).bool()
-    }
+    time=250
 )
 print(connection.pipeline[0].value)