Move monitors to device

BindsNET · Feb 17, 2021 · 5ee9497 · 5ee9497
1 parent d58b512
commit 5ee9497
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mnist/reservoir.py b/examples/mnist/reservoir.py
@@ -86,10 +86,10 @@
 
 spikes = {}
 for l in network.layers:
-    spikes[l] = Monitor(network.layers[l], ["s"], time=time)
+    spikes[l] = Monitor(network.layers[l], ["s"], time=time, device=device)
     network.add_monitor(spikes[l], name="%s_spikes" % l)
 
-voltages = {"O": Monitor(network.layers["O"], ["v"], time=time)}
+voltages = {"O": Monitor(network.layers["O"], ["v"], time=time, device=device)}
 network.add_monitor(voltages["O"], name="O_voltages")
 
 # Directs network to GPU