diff --git a/train.py b/train.py index ff86d4c15c..929a3f0519 100644 --- 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,