From fedc2150b3f39f743f25c50d793c3f52bf85076a Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 14 Nov 2019 17:12:55 -0800 Subject: [PATCH] updates --- train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index e5979645..3350f3a8 100644 --- a/train.py +++ b/train.py @@ -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': 40.0, # obj loss gain (*=img_size/320 if img_size != 320) + 'obj': 64.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) @@ -419,8 +419,8 @@ if __name__ == '__main__': print(opt) device = torch_utils.select_device(opt.device, apex=mixed_precision) - # scale hyp['obj'] by img_size (evolved at 320) - hyp['obj'] *= opt.img_size / 320. + # scale hyp['obj'] by img_size (evolved at 512) + hyp['obj'] *= opt.img_size / 512. tb_writer = None if not opt.evolve: # Train normally