glenn-jocher commited on
Commit
8e5c665
·
1 Parent(s): 41523e2

update train.py remove save_json final_epoch

Browse files
Files changed (1) hide show
  1. train.py +0 -1
train.py CHANGED
@@ -306,7 +306,6 @@ def train(hyp, opt, device, tb_writer=None):
306
  results, maps, times = test.test(opt.data,
307
  batch_size=total_batch_size,
308
  imgsz=imgsz_test,
309
- save_json=final_epoch and opt.data.endswith(os.sep + 'coco.yaml'),
310
  model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema,
311
  single_cls=opt.single_cls,
312
  dataloader=testloader,
 
306
  results, maps, times = test.test(opt.data,
307
  batch_size=total_batch_size,
308
  imgsz=imgsz_test,
 
309
  model=ema.ema.module if hasattr(ema.ema, 'module') else ema.ema,
310
  single_cls=opt.single_cls,
311
  dataloader=testloader,