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

updates

This commit is contained in:
Glenn Jocher 2019-11-09 10:56:38 -08:00
parent 97ac36ec6c
commit 579fdc57f8
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

@ -60,9 +60,6 @@ def train():
hyp['cls_pw'] = 1. hyp['cls_pw'] = 1.
hyp['obj_pw'] = 1. hyp['obj_pw'] = 1.
if img_size > 320: # scale hyp['obj'] by img_size (evolved at 320)
hyp['obj'] *= img_size / 320.
# Initialize # Initialize
init_seeds() init_seeds()
multi_scale = opt.multi_scale multi_scale = opt.multi_scale
@ -422,6 +419,9 @@ if __name__ == '__main__':
print(opt) print(opt)
device = torch_utils.select_device(opt.device, apex=mixed_precision) 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.
tb_writer = None tb_writer = None
if not opt.evolve: # Train normally if not opt.evolve: # Train normally
try: try: