Skip to content

Commit 20a702d

Browse files
authored
Update config.py
default value of `epochs` set to 3
1 parent ec191ea commit 20a702d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
num_synthetic_imgs = 18000
2222
train_percentage = 0.8
2323

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

0 commit comments

Comments
 (0)