未验证 提交 e5d59918 编写于 作者: L LielinJiang 提交者: GitHub

fix save img bug (#344)

上级 a4a54e51
......@@ -98,6 +98,7 @@ class Trainer:
# build metrics
self.metrics = None
self.is_save_img = True
validate_cfg = cfg.get('validate', None)
if validate_cfg and 'metrics' in validate_cfg:
self.metrics = self.model.setup_metrics(validate_cfg['metrics'])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册