未验证 提交 2f9e086a 编写于 作者: S shangliang Xu 提交者: GitHub

silence ema save log (#5305)

上级 161d5bbe
...@@ -232,4 +232,4 @@ def save_model(model, save_dir, save_name, last_epoch, optimizer=None): ...@@ -232,4 +232,4 @@ def save_model(model, save_dir, save_name, last_epoch, optimizer=None):
state_dict = optimizer.state_dict() state_dict = optimizer.state_dict()
state_dict['last_epoch'] = last_epoch state_dict['last_epoch'] = last_epoch
paddle.save(state_dict, save_path + ".pdopt") paddle.save(state_dict, save_path + ".pdopt")
logger.info("Save checkpoint: {}".format(save_dir)) logger.info("Save checkpoint: {}".format(save_dir))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册