Add amp to static fix (#543)

* add amp and to_static * fix amp and to_static
PaddlePaddle · Nov 29, 2024 · e163352 · e163352
1 parent 6b09113
commit e163352
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/paddlets/models/anomaly/dl/anomaly_base.py b/paddlets/models/anomaly/dl/anomaly_base.py
@@ -1006,7 +1006,7 @@ def save(self,
         network = self._network
         callback_container = self._callback_container
         loss_fn = self._loss_fn
-        if self.use_amp:
+        if hasattr(self, 'use_amp') and self.use_amp:
             scaler = self.scaler
             self.scaler = None
 
@@ -1034,8 +1034,8 @@ def save(self,
         self._network = network
         self._callback_container = callback_container
         self._loss_fn = loss_fn
-        if self.use_amp:
-            scaler = self.scaler
+        if hasattr(self, 'use_amp') and self.use_amp:
+            self.scaler = scaler
         return
 
     @staticmethod

diff --git a/paddlets/models/classify/dl/paddle_base.py b/paddlets/models/classify/dl/paddle_base.py
@@ -831,7 +831,7 @@ def save(self,
         network = self._network
         callback_container = self._callback_container
         loss_fn = self._loss_fn
-        if self.use_amp:
+        if hasattr(self, 'use_amp') and self.use_amp:
             scaler = self.scaler
             self.scaler = None
 
@@ -859,7 +859,7 @@ def save(self,
         self._network = network
         self._callback_container = callback_container
         self._loss_fn = loss_fn
-        if self.use_amp:
+        if hasattr(self, 'use_amp') and self.use_amp:
             self.scaler = scaler
         return
 

diff --git a/paddlets/models/forecasting/dl/paddle_base.py b/paddlets/models/forecasting/dl/paddle_base.py
@@ -255,7 +255,7 @@ def save(self,
         network = self._network
         callback_container = self._callback_container
         loss_fn = self._loss_fn
-        if self.use_amp:
+        if hasattr(self, 'use_amp') and self.use_amp:
             scaler = self.scaler
             self.scaler = None
 
@@ -283,9 +283,8 @@ def save(self,
         self._network = network
         self._callback_container = callback_container
         self._loss_fn = loss_fn
-        if self.use_amp:
+        if hasattr(self, 'use_amp') and self.use_amp:
             self.scaler = scaler
-
         return
 
     @staticmethod