add default value for label

2021-06-07 09:43:47 +08:00 · 2021-06-07 09:43:47 +08:00 · e8959c31d5
parent 03d32d61e0
commit e8959c31d5
1 changed files with 1 additions and 1 deletions
--- a/tools/export_model.py
+++ b/tools/export_model.py
@ -66,7 +66,7 @@ class IdentityHead(nn.Layer):
    def __init__(self):
        super(IdentityHead, self).__init__()

-    def forward(self, x, label):
+    def forward(self, x, label=None):
        return {"features": x, "logits": None}