From 296ba94c1c34455227fafe30e1bd97a5a9a717c7 Mon Sep 17 00:00:00 2001
From: weishengyu <weishengyu@baidu.com>
Date: Mon, 7 Jun 2021 15:05:26 +0800
Subject: [PATCH] dbg

---
 ppcls/arch/gears/identity_head.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ppcls/arch/gears/identity_head.py b/ppcls/arch/gears/identity_head.py
index d4343b9c..0d30d0e4 100644
--- a/ppcls/arch/gears/identity_head.py
+++ b/ppcls/arch/gears/identity_head.py
@@ -1,3 +1,6 @@
+from paddle import nn
+
+
 class IdentityHead(nn.Layer):
     def __init__(self):
         super(IdentityHead, self).__init__()
-- 
GitLab