提交 8e5c6657 authored 作者: Glenn Jocher's avatar Glenn Jocher

update train.py remove save_json final_epoch

上级 41523e2c
......@@ -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,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论