fix distill error (#6907)

3b36a655 · Guanghua Yu · GitHub · 9a1ee55c · 3b36a655
隐藏空白更改
内联并排

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
@@ -262,7 +262,7 @@ class FGDFeatureLoss(nn.Layer):
        zeros_init = parameter_init("constant", 0.0)

        if student_channels != teacher_channels:
-            self.align = nn.Conv2d(
+            self.align = nn.Conv2D(
                student_channels,
                teacher_channels,
                kernel_size=1,