We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3008d88 commit 7aac812Copy full SHA for 7aac812
tutorials/02-intermediate/deep_residual_network/main.py
@@ -16,6 +16,7 @@
16
17
# Hyper-parameters
18
num_epochs = 80
19
+batch_size = 100
20
learning_rate = 0.001
21
22
# Image preprocessing modules
@@ -37,11 +38,11 @@
37
38
39
# Data loader
40
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
- batch_size=100,
41
+ batch_size=batch_size,
42
shuffle=True)
43
44
test_loader = torch.utils.data.DataLoader(dataset=test_dataset,
45
46
shuffle=False)
47
48
# 3x3 convolution
0 commit comments