diff --git a/train.py b/train.py index 108c926..86a9dbc 100644 --- a/train.py +++ b/train.py @@ -306,7 +306,6 @@ def train(hyp, opt, device, tb_writer=None): results, maps, times = test.test(opt.data, batch_size=total_batch_size, imgsz=imgsz_test, - save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'), model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema, single_cls=opt.single_cls, dataloader=testloader,