mnist_forward_forward: Load dataset from comment path

Rtoax · Rtoax · commit 85d44fc12999 · 2025-08-27T10:03:02.000+08:00
Load the same dataset file as other MNIST test cases.

Signed-off-by: Rong Tao &lt;rongtao@cestc.cn&gt;
diff --git a/mnist_forward_forward/main.py b/mnist_forward_forward/main.py
@@ -154,10 +154,10 @@ def train(self, x_pos, x_neg):
         ]
     )
     train_loader = DataLoader(
-        MNIST("./data/", train=True, download=True, transform=transform), **train_kwargs
+        MNIST("../data/", train=True, download=True, transform=transform), **train_kwargs
     )
     test_loader = DataLoader(
-        MNIST("./data/", train=False, download=True, transform=transform), **test_kwargs
+        MNIST("../data/", train=False, download=True, transform=transform), **test_kwargs
     )
     net = Net([784, 500, 500])
     x, y = next(iter(train_loader))

Original file line number	Diff line number	Diff line change
`@@ -154,10 +154,10 @@ def train(self, x_pos, x_neg):`
`154`	`154`	`]`
`155`	`155`	`)`
`156`	`156`	`train_loader = DataLoader(`
`157`		`- MNIST("./data/", train=True, download=True, transform=transform), **train_kwargs`
	`157`	`+ MNIST("../data/", train=True, download=True, transform=transform), **train_kwargs`
`158`	`158`	`)`
`159`	`159`	`test_loader = DataLoader(`
`160`		`- MNIST("./data/", train=False, download=True, transform=transform), **test_kwargs`
	`160`	`+ MNIST("../data/", train=False, download=True, transform=transform), **test_kwargs`
`161`	`161`	`)`
`162`	`162`	`net = Net([784, 500, 500])`
`163`	`163`	`x, y = next(iter(train_loader))`