Skip to content

P
PaddleDetection

s920243400 / PaddleDetection
与 Fork 源项目一致

Fork自 PaddlePaddle / PaddleDetection

通知 2
Star 0
Fork 0
P
PaddleDetection
未验证 提交 da3a1fc9 编写于 作者: U user1018 提交者： GitHub
浏览文件
操作

fix distill bug (#6733)

上级 4708b081
显示空白变更内容
内联 并排
Showing with 1 addition and 1 deletion
ppdet/slim/distill.py
浏览文件 @ da3a1fc9
...@@ -436,7 +436,7 @@ class FGDFeatureLoss(nn.Layer): ...@@ -436,7 +436,7 @@ class FGDFeatureLoss(nn.Layer):
Mask_bg = paddle.ones_like(tea_spatial_att) Mask_bg = paddle.ones_like(tea_spatial_att)
one_tmp = paddle.ones([*tea_spatial_att.shape[1:]]) one_tmp = paddle.ones([*tea_spatial_att.shape[1:]])
zero_tmp = paddle.zeros([*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 Mask_bg.stop_gradient = True
one_tmp.stop_gradient = True one_tmp.stop_gradient = True
zero_tmp.stop_gradient = True zero_tmp.stop_gradient = True
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或