updates

2019-12-27 10:52:19 -08:00 · 2019-12-27 10:52:19 -08:00 · 56d7261083
commit 56d7261083
parent 2cc805edda
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -330,7 +330,7 @@ def train():
                                      img_size=opt.img_size,
                                      model=model,
                                      conf_thres=0.001 if final_epoch else 0.1,  # 0.1 for speed
-                                      iou_thres=0.6 if opt.evolve else 0.5,
+                                      iou_thres=0.6 if final_epoch and is_coco else 0.5,
                                      save_json=final_epoch and is_coco,
                                      dataloader=testloader)