Fix inconsistent evaluation params

Following: ultralytics/yolov5#3934
eldarkurtic · Jul 9, 2021 · 34e80f3 · 34e80f3
1 parent e3bbbad
commit 34e80f3
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -441,8 +441,6 @@ def train(hyp, opt, device, tb_writer=None):
                 results, _, _ = test.test(opt.data,
                                           batch_size=batch_size * 2,
                                           imgsz=imgsz_test,
-                                          conf_thres=0.001,
-                                          iou_thres=0.7,
                                           model=test_model,
                                           single_cls=opt.single_cls,
                                           dataloader=testloader,