fix distill bug (#6733)

da3a1fc9 · user1018 · GitHub · 4708b081 · da3a1fc9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

ppdet/slim/distill.py ppdet/slim/distill.py +1 -1

未找到文件。
--- 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