dbg

2021-06-07 15:05:26 +08:00 · 2021-06-07 15:05:26 +08:00 · 296ba94c1c
parent 07089dcfe7
commit 296ba94c1c
1 changed files with 3 additions and 0 deletions
--- 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__()