updates
This commit is contained in:
parent
a1d5f62334
commit
7807337c5f
2
train.py
2
train.py
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue