From 9ab1ca3f0ce291da45b617a6377d9f3c4f45882c Mon Sep 17 00:00:00 2001
From: weishengyu <weisy11@163.com>
Date: Thu, 7 Apr 2022 18:31:45 +0800
Subject: [PATCH] dbg

---
 ppcls/arch/gears/bnneck.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/ppcls/arch/gears/bnneck.py b/ppcls/arch/gears/bnneck.py
index 8e97409e..15b6f46b 100644
--- a/ppcls/arch/gears/bnneck.py
+++ b/ppcls/arch/gears/bnneck.py
@@ -1,4 +1,7 @@
-class BNNeck(nn.Layer):
+import paddle
+
+
+class BNNeck(paddle.nn.Layer):
     def __init__(self, num_filters, trainable=False):
         super(BNNeck, self).__init__()
         self.num_filters = num_filters
-- 
GitLab