Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
glenn-jocher 2019-06-30 17:47:10 +02:00
parent db2674aa31
commit 32f5ea955b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
train.py
View File

@ -47,9 +47,9 @@ def train(
data_cfg, data_cfg,
img_size=416, img_size=416,
resume=False, resume=False,
epochs=100, # 500200 batches at bs 4, 117263 images = 68 epochs epochs=100, # 500200 batches at bs 16, 117263 images = 273 epochs
batch_size=16, batch_size=8,
accumulate=4, # effective bs = 64 = batch_size * accumulate accumulate=8, # effective bs = batch_size * accumulate = 8 * 8 = 64
freeze_backbone=False, freeze_backbone=False,
transfer=False # Transfer learning (train only YOLO layers) transfer=False # Transfer learning (train only YOLO layers)
): ):