This commit is contained in:
Glenn Jocher 2019-05-31 14:30:27 +02:00
parent a1d5f62334
commit 7807337c5f
1 changed files with 1 additions and 1 deletions

View File

@ -61,7 +61,7 @@ def train(
data_cfg, data_cfg,
img_size=416, img_size=416,
resume=False, resume=False,
epochs=68, # 500200 batches at bs 4, dataset length 117263 epochs=100, # 500200 batches at bs 4, 117263 images = 68 epochs
batch_size=16, batch_size=16,
accumulate=4, # effective bs = 64 = batch_size * accumulate accumulate=4, # effective bs = 64 = batch_size * accumulate
multi_scale=False, multi_scale=False,