From 85d44fc129994daf0147972ce615ebaf0eb4d272 Mon Sep 17 00:00:00 2001 From: Rong Tao Date: Wed, 27 Aug 2025 10:00:17 +0800 Subject: [PATCH] mnist_forward_forward: Load dataset from comment path Load the same dataset file as other MNIST test cases. Signed-off-by: Rong Tao --- mnist_forward_forward/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mnist_forward_forward/main.py b/mnist_forward_forward/main.py index 25bc187b29..7b651d3083 100644 --- 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))