diff --git a/paddlespeech/s2t/training/trainer.py b/paddlespeech/s2t/training/trainer.py index 53a5d03f7..d9ac6f8b0 100644 --- a/paddlespeech/s2t/training/trainer.py +++ b/paddlespeech/s2t/training/trainer.py @@ -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,