From 6bc855d8b6491715b51101d34a3ee8b6ccd1944c Mon Sep 17 00:00:00 2001
From: zhaoyingli <86812880+zhaoyinglia@users.noreply.github.com>
Date: Thu, 29 Sep 2022 16:50:23 +0800
Subject: [PATCH] [AutoParallel] fix amp when predict (#46637)

---
 python/paddle/distributed/passes/auto_parallel_amp.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/paddle/distributed/passes/auto_parallel_amp.py b/python/paddle/distributed/passes/auto_parallel_amp.py
index 064075bff3..dcfac246f4 100644
--- 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:
-- 
GitLab