Unverified 提交 89a6e9ff authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Compute loss on final val (#5017)

上级 b20e3814
...@@ -419,7 +419,8 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary ...@@ -419,7 +419,8 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
save_json=is_coco, save_json=is_coco,
verbose=True, verbose=True,
plots=True, plots=True,
callbacks=callbacks) # val best model with plots callbacks=callbacks,
compute_loss=compute_loss) # val best model with plots
callbacks.run('on_train_end', last, best, plots, epoch) callbacks.run('on_train_end', last, best, plots, epoch)
LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}") LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论