diff --git a/paddlespeech/s2t/training/trainer.py b/paddlespeech/s2t/training/trainer.py index 53a5d03f781375b3de2bd529bd0b54ba9ac9ede6..d9ac6f8b0f3eef2e4d0def60ae7b4b6fa245be1a 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,