From 56d7261083286850cf021ab603d60f8321a51951 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 27 Dec 2019 10:52:19 -0800
Subject: [PATCH] updates

---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 26e6f2b8..5fb31e25 100644
--- 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)