Skip to content

Commit 7aac812

Browse files
committed
minor refactoring for batch size in deep residual network
1 parent 3008d88 commit 7aac812

File tree

1 file changed

+3
-2
lines changed
  • tutorials/02-intermediate/deep_residual_network

1 file changed

+3
-2
lines changed

tutorials/02-intermediate/deep_residual_network/main.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
# Hyper-parameters
1818
num_epochs = 80
19+
batch_size = 100
1920
learning_rate = 0.001
2021

2122
# Image preprocessing modules
@@ -37,11 +38,11 @@
3738

3839
# Data loader
3940
train_loader = torch.utils.data.DataLoader(dataset=train_dataset,
40-
batch_size=100,
41+
batch_size=batch_size,
4142
shuffle=True)
4243

4344
test_loader = torch.utils.data.DataLoader(dataset=test_dataset,
44-
batch_size=100,
45+
batch_size=batch_size,
4546
shuffle=False)
4647

4748
# 3x3 convolution

0 commit comments

Comments
 (0)