Skip to content

Commit

Permalink
Update batch_eth_mnist.py
Browse files Browse the repository at this point in the history
Move monitors to device variable
  • Loading branch information
Hananel-Hazan committed Feb 11, 2021
1 parent 3a296e6 commit bdeaa03
Showing 1 changed file with 6 additions and 4 deletions.
10 changes: 6 additions & 4 deletions examples/mnist/batch_eth_mnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -133,23 +133,25 @@
accuracy = {"all": [], "proportion": []}

# Voltage recording for excitatory and inhibitory layers.
exc_voltage_monitor = Monitor(network.layers["Ae"], ["v"], time=int(time / dt))
inh_voltage_monitor = Monitor(network.layers["Ai"], ["v"], time=int(time / dt))
exc_voltage_monitor = Monitor(network.layers["Ae"], ["v"], time=int(time / dt), device=device)
inh_voltage_monitor = Monitor(network.layers["Ai"], ["v"], time=int(time / dt), device=device)
network.add_monitor(exc_voltage_monitor, name="exc_voltage")
network.add_monitor(inh_voltage_monitor, name="inh_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)

Expand Down

0 comments on commit bdeaa03

Please sign in to comment.