diff --git a/train.py b/train.py index f37eb77b..9df15801 100644 --- a/train.py +++ b/train.py @@ -315,8 +315,6 @@ def train(): batch_size=batch_size, img_size=img_size_test, model=ema.ema, - conf_thres=0.001 if final_epoch else 0.01, # 0.001 for best mAP, 0.01 for speed - iou_thres=0.6, save_json=final_epoch and is_coco, single_cls=opt.single_cls, dataloader=testloader)