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.
batch_size
1 parent 55cb521 commit 1560c64Copy full SHA for 1560c64
config.py
@@ -25,7 +25,7 @@
25
train_percentage = 0.8
26
27
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
+ckpt='model01.pth' # model file path to load the weights from, only useful when resume is True
29
lr = 3e-6 # learning rate
30
epochs = 3 # epochs to train for
31
0 commit comments