This commit is contained in:
Glenn Jocher 2019-11-18 12:01:17 -08:00
parent 7ebb7d1310
commit 2ba1a4c9cc
1 changed files with 3 additions and 3 deletions

View File

@ -24,7 +24,7 @@ results_file = 'results.txt'
hyp = {'giou': 3.31, # giou loss gain
'cls': 42.4, # cls loss gain
'cls_pw': 1.0, # cls BCELoss positive_weight
'obj': 64.0, # obj loss gain (*=img_size/320 if img_size != 320)
'obj': 52.0, # obj loss gain (*=img_size/320 if img_size != 320)
'obj_pw': 1.0, # obj BCELoss positive_weight
'iou_t': 0.213, # iou training threshold
'lr0': 0.00261, # initial learning rate (SGD=1E-3, Adam=9E-5)
@ -417,8 +417,8 @@ if __name__ == '__main__':
print(opt)
device = torch_utils.select_device(opt.device, apex=mixed_precision)
# scale hyp['obj'] by img_size (evolved at 512)
hyp['obj'] *= opt.img_size / 512.
# scale hyp['obj'] by img_size (evolved at 416)
hyp['obj'] *= opt.img_size / 416.
tb_writer = None
if not opt.evolve: # Train normally