“ddf6b722e788625c62a38ee62042905a4c40cd08”上不存在“mobile/src/operators/scale_op.cpp”
提交 2f4414a5 编写于 作者: Z zxcd

fix scaler save

上级 fbd27aab
...@@ -189,7 +189,7 @@ class Trainer(): ...@@ -189,7 +189,7 @@ class Trainer():
"step": self.iteration, "step": self.iteration,
"epoch": self.epoch, "epoch": self.epoch,
"lr": self.optimizer.get_lr(), "lr": self.optimizer.get_lr(),
"scaler": self.scaler "scaler": self.scaler.state_dict()
}) })
self.checkpoint.save_parameters(self.checkpoint_dir, self.iteration self.checkpoint.save_parameters(self.checkpoint_dir, self.iteration
if tag is None else tag, self.model, if tag is None else tag, self.model,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册