summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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)