diff --git a/examples/mnist/batch_eth_mnist.py b/examples/mnist/batch_eth_mnist.py index 9f28eb9b..103692e6 100644 --- a/examples/mnist/batch_eth_mnist.py +++ b/examples/mnist/batch_eth_mnist.py @@ -45,7 +45,7 @@ parser.add_argument("--test", dest="train", action="store_false") parser.add_argument("--plot", dest="plot", action="store_true") parser.add_argument("--gpu", dest="gpu", action="store_true") -parser.set_defaults(plot=False, gpu=False) +parser.set_defaults(plot=True, gpu=False) args = parser.parse_args() diff --git a/examples/mnist/conv_mnist.py b/examples/mnist/conv_mnist.py index 0a29f9bd..1656d6a3 100644 --- a/examples/mnist/conv_mnist.py +++ b/examples/mnist/conv_mnist.py @@ -41,7 +41,7 @@ parser.add_argument("--test", dest="train", action="store_false") parser.add_argument("--plot", dest="plot", action="store_true") parser.add_argument("--gpu", dest="gpu", action="store_true") -parser.set_defaults(plot=False, gpu=False, train=True) +parser.set_defaults(plot=True, gpu=False, train=True) args = parser.parse_args() diff --git a/examples/mnist/eth_mnist.py b/examples/mnist/eth_mnist.py index 11771eb4..b6ba50b7 100644 --- a/examples/mnist/eth_mnist.py +++ b/examples/mnist/eth_mnist.py @@ -44,7 +44,7 @@ parser.add_argument("--test", dest="train", action="store_false") parser.add_argument("--plot", dest="plot", action="store_true") parser.add_argument("--gpu", dest="gpu", action="store_true") -parser.set_defaults(plot=False, gpu=False) +parser.set_defaults(plot=True, gpu=False) args = parser.parse_args() diff --git a/examples/mnist/reservoir.py b/examples/mnist/reservoir.py index 2e2340fb..455d9399 100644 --- a/examples/mnist/reservoir.py +++ b/examples/mnist/reservoir.py @@ -146,12 +146,12 @@ ims=inpt_ims, ) spike_ims, spike_axes = plot_spikes( - {layer: spikes[layer].get("s").view(-1, time) for layer in spikes}, + {layer: spikes[layer].get("s").view(time, -1) for layer in spikes}, axes=spike_axes, ims=spike_ims, ) voltage_ims, voltage_axes = plot_voltages( - {layer: voltages[layer].get("v").view(-1, time) for layer in voltages}, + {layer: voltages[layer].get("v").view(time, -1) for layer in voltages}, ims=voltage_ims, axes=voltage_axes, ) diff --git a/examples/mnist/supervised_mnist.py b/examples/mnist/supervised_mnist.py index 3722da12..390ab123 100644 --- a/examples/mnist/supervised_mnist.py +++ b/examples/mnist/supervised_mnist.py @@ -42,7 +42,7 @@ parser.add_argument("--plot", dest="plot", action="store_true") parser.add_argument("--gpu", dest="gpu", action="store_true") parser.add_argument("--device_id", type=int, default=0) -parser.set_defaults(plot=False, gpu=False, train=True) +parser.set_defaults(plot=True, gpu=False, train=True) args = parser.parse_args()