“4b71be3a93f7dafc19373f8a63259ac0f68ab913”上不存在“_all_pages/v2.1.0/source_compile.md”
提交 9092594f 编写于 作者: H Hui Zhang

resume not increase epoch and step, before_train do it

上级 ad6d09d6
...@@ -199,8 +199,8 @@ class Trainer(): ...@@ -199,8 +199,8 @@ class Trainer():
if infos: if infos:
# just restore ckpt # just restore ckpt
# lr will resotre from optimizer ckpt # lr will resotre from optimizer ckpt
self.iteration = infos["step"] + 1 self.iteration = infos["step"]
self.epoch = infos["epoch"] + 1 self.epoch = infos["epoch"]
scratch = False scratch = False
logger.info( logger.info(
f"Restore ckpt: epoch {self.epoch }, step {self.iteration}!") 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.
先完成此消息的编辑!
想要评论请 注册