Merge pull request #755 from andreapignaz/fix/eth_mnist

Hananel-Hazan · web-flow · commit 5b8618ac7341 · 2026-05-07T14:11:48.000-04:00
Fix eth_mnist example
diff --git a/examples/mnist/eth_mnist.py b/examples/mnist/eth_mnist.py
@@ -253,7 +253,7 @@
         if plot:
             image = batch["image"].view(28, 28)
             inpt = inputs["X"].view(time, 784).sum(0).view(28, 28)
-            input_exc_weights = network.connections[("X", "Ae")].w
+            input_exc_weights = network.connections[("X", "Ae")].pipeline[0].value
             square_weights = get_square_weights(
                 input_exc_weights.view(784, n_neurons), n_sqrt, 28
             )

Original file line number	Diff line number	Diff line change
`@@ -253,7 +253,7 @@`
`253`	`253`	`if plot:`
`254`	`254`	`image = batch["image"].view(28, 28)`
`255`	`255`	`inpt = inputs["X"].view(time, 784).sum(0).view(28, 28)`
`256`		`- input_exc_weights = network.connections[("X", "Ae")].w`
	`256`	`+ input_exc_weights = network.connections[("X", "Ae")].pipeline[0].value`
`257`	`257`	`square_weights = get_square_weights(`
`258`	`258`	`input_exc_weights.view(784, n_neurons), n_sqrt, 28`
`259`	`259`	`)`