提交 9092594f 编写于 作者: H Hui Zhang

resume not increase epoch and step, before_train do it

上级 ad6d09d6
......@@ -199,8 +199,8 @@ class Trainer():
if infos:
# just restore ckpt
# lr will resotre from optimizer ckpt
self.iteration = infos["step"] + 1
self.epoch = infos["epoch"] + 1
self.iteration = infos["step"]
self.epoch = infos["epoch"]
scratch = False
logger.info(
f"Restore ckpt: epoch {self.epoch }, step {self.iteration}!")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册