Skip to content

Commit 5ee9497

Browse files
Move monitors to device
1 parent d58b512 commit 5ee9497

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/mnist/reservoir.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,10 +86,10 @@
8686

8787
spikes = {}
8888
for l in network.layers:
89-
spikes[l] = Monitor(network.layers[l], ["s"], time=time)
89+
spikes[l] = Monitor(network.layers[l], ["s"], time=time, device=device)
9090
network.add_monitor(spikes[l], name="%s_spikes" % l)
9191

92-
voltages = {"O": Monitor(network.layers["O"], ["v"], time=time)}
92+
voltages = {"O": Monitor(network.layers["O"], ["v"], time=time, device=device)}
9393
network.add_monitor(voltages["O"], name="O_voltages")
9494

9595
# Directs network to GPU

0 commit comments

Comments
 (0)