Skip to content

Commit 55cb521

Browse files
authored
changed the default batch_size to 2
1 parent 81c7bd3 commit 55cb521

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
epochs = 3 # epochs to train for
3131

3232
# batch size for train and val loaders
33-
batch_size = 4
33+
batch_size = 32 # try decreasing the batch_size if there is a memory error
3434

3535
# log interval for training and validation
3636
log_interval = 25

0 commit comments

Comments
 (0)