diff --git a/ppcls/arch/gears/identity_head.py b/ppcls/arch/gears/identity_head.py
index 0d30d0e45..7d11e5742 100644
--- a/ppcls/arch/gears/identity_head.py
+++ b/ppcls/arch/gears/identity_head.py
@@ -6,4 +6,4 @@ class IdentityHead(nn.Layer):
         super(IdentityHead, self).__init__()
 
     def forward(self, x, label=None):
-        return {"features": x, "logits": None}
\ No newline at end of file
+        return {"features": x, "logits": None}