Skip to content

Commit 1560c64

Browse files
authored
changed the default batch_size to 32
1 parent 55cb521 commit 1560c64

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
@@ -25,7 +25,7 @@
2525
train_percentage = 0.8
2626

2727
resume = not True # False for trainig from scratch, True for loading a previously saved weight
28-
ckpt='model08.pth' # model file path to load the weights from, only useful when resume is True
28+
ckpt='model01.pth' # model file path to load the weights from, only useful when resume is True
2929
lr = 3e-6 # learning rate
3030
epochs = 3 # epochs to train for
3131

0 commit comments

Comments
 (0)