Update SOM_LM-SNNs.py

Move monitor to device variable
BindsNET · Feb 11, 2021 · 3c2e550 · 3c2e550
1 parent caf0051
commit 3c2e550
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/mnist/SOM_LM-SNNs.py b/examples/mnist/SOM_LM-SNNs.py
@@ -120,21 +120,21 @@
 accuracy = {"all": [], "proportion": []}
 
 # Voltage recording for excitatory and inhibitory layers.
-som_voltage_monitor = Monitor(network.layers["Y"], ["v"], time=int(time / dt))
+som_voltage_monitor = Monitor(network.layers["Y"], ["v"], time=int(time / dt), device=device)
 network.add_monitor(som_voltage_monitor, name="som_voltage")
 
 # Set up monitors for spikes and voltages
 spikes = {}
 for layer in set(network.layers):
     spikes[layer] = Monitor(
-        network.layers[layer], state_vars=["s"], time=int(time / dt)
+        network.layers[layer], state_vars=["s"], time=int(time / dt), device=device
     )
     network.add_monitor(spikes[layer], name="%s_spikes" % layer)
 
 voltages = {}
 for layer in set(network.layers) - {"X"}:
     voltages[layer] = Monitor(
-        network.layers[layer], state_vars=["v"], time=int(time / dt)
+        network.layers[layer], state_vars=["v"], time=int(time / dt), device=device
     )
     network.add_monitor(voltages[layer], name="%s_voltages" % layer)