From 309151378d608a1710ec7217f06b81d72b75164e Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Fri, 9 Sep 2022 10:39:47 +0800
Subject: [PATCH] fix distill error (#6906)

---
 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 996342e15..a20c93a31 100644
--- 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,
-- 
GitLab