fix format

2021-06-07 15:05:51 +08:00 · 2021-06-07 15:05:51 +08:00 · 25f6af99b2
parent 296ba94c1c
commit 25f6af99b2
1 changed files with 1 additions and 1 deletions
--- 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}
+        return {"features": x, "logits": None}