提交 2f4414a5 编写于 作者: Z zxcd

fix scaler save

上级 fbd27aab
......@@ -189,7 +189,7 @@ class Trainer():
"step": self.iteration,
"epoch": self.epoch,
"lr": self.optimizer.get_lr(),
"scaler": self.scaler
"scaler": self.scaler.state_dict()
})
self.checkpoint.save_parameters(self.checkpoint_dir, self.iteration
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.
先完成此消息的编辑!
想要评论请 注册