[AutoParallel] fix amp when predict (#46637)

6bc855d8 · zhaoyingli · GitHub · 09569323 · 6bc855d8
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

python/paddle/distributed/passes/auto_parallel_amp.py python/paddle/distributed/passes/auto_parallel_amp.py +1 -0

未找到文件。
--- a/python/paddle/distributed/passes/auto_parallel_amp.py
+++ b/python/paddle/distributed/passes/auto_parallel_amp.py
@@ -520,6 +520,7 @@ class AMPPass(PassBase):
        self._loss_scaling = None
        self._num_good_steps = None
        self._num_bad_steps = None
+        self._loss = None

    def _check_self(self):
        if self.get_attr("init_loss_scaling") < 0: