fix scaler save

2 years ago · 2f4414a5f8
parent fbd27aab41
commit 2f4414a5f8
1 changed files with 1 additions and 1 deletions
--- 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,