From 2f4414a5f8386aab92e807719572a80df7da74bd Mon Sep 17 00:00:00 2001
From: zxcd <228587199@qq.com>
Date: Mon, 17 Apr 2023 03:32:20 +0000
Subject: [PATCH] fix scaler save

---
 paddlespeech/s2t/training/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,