diff --git a/ppcls/arch/gears/identity_head.py b/ppcls/arch/gears/identity_head.py index 8542dd730e517b0d36b601db1f3a16d6adc1dd39..7d11e5742e1645a4a3407c73f1c6d316e0aa4e6d 100644 --- a/ppcls/arch/gears/identity_head.py +++ b/ppcls/arch/gears/identity_head.py @@ -1,5 +1,6 @@ from paddle import nn + class IdentityHead(nn.Layer): def __init__(self): super(IdentityHead, self).__init__()