From da3a1fc995d5e0f35a0d86c222ca9845f62c9d82 Mon Sep 17 00:00:00 2001
From: user1018 <614803115@qq.com>
Date: Wed, 24 Aug 2022 10:17:08 +0800
Subject: [PATCH] fix distill bug (#6733)

---
 ppdet/slim/distill.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/slim/distill.py b/ppdet/slim/distill.py
index 8f7ee7412..996342e15 100644
--- a/ppdet/slim/distill.py
+++ b/ppdet/slim/distill.py
@@ -436,7 +436,7 @@ class FGDFeatureLoss(nn.Layer):
         Mask_bg = paddle.ones_like(tea_spatial_att)
         one_tmp = paddle.ones([*tea_spatial_att.shape[1:]])
         zero_tmp = paddle.zeros([*tea_spatial_att.shape[1:]])
-        mask_fg.stop_gradient = True
+        Mask_fg.stop_gradient = True
         Mask_bg.stop_gradient = True
         one_tmp.stop_gradient = True
         zero_tmp.stop_gradient = True
-- 
GitLab