index
:
pytorch
master
PyTorch experiments
summary
refs
log
tree
commit
diff
log msg
author
committer
range
diff options
context:
1
2
3
4
5
6
7
8
9
10
15
20
25
30
35
40
space:
include
ignore
mode:
unified
ssdiff
stat only
Diffstat
-rw-r--r--
mnist.py
1
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)