update train.py remove save_json final_epoch

ultralytics · Aug 10, 2020 · 8e5c665 · 8e5c665
1 parent 41523e2
commit 8e5c665
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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,