summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAnthony Wang2021-08-31 12:22:37 -0500
committerAnthony Wang2021-08-31 12:22:37 -0500
commit1944c4a9613c15bde2eba95b4ab4fada88c19cce (patch)
tree8e1313d43713fd6ca9843a7df3cc50394f50403b
parent2b8c1d706a16c7bf93df76e08988e906f39881d1 (diff)
Add latest files
-rw-r--r--mnist.py1
1 files changed, 1 insertions, 0 deletions
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)