From 1944c4a9613c15bde2eba95b4ab4fada88c19cce Mon Sep 17 00:00:00 2001 From: Anthony Wang Date: Tue, 31 Aug 2021 12:22:37 -0500 Subject: Add latest files --- mnist.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mnist.py b/mnist.py index f6aba1d..d6bed0f 100644 --- a/mnist.py +++ b/mnist.py @@ -21,6 +21,7 @@ test_data = datasets.MNIST( transform=ToTensor(), ) + batch_size = 100 train_loader = DataLoader(training_data, batch_size=batch_size) -- cgit v1.2.3-70-g09d2