提交 e8959c31 编写于 作者: W weishengyu

add default value for label

上级 03d32d61
...@@ -66,7 +66,7 @@ class IdentityHead(nn.Layer): ...@@ -66,7 +66,7 @@ class IdentityHead(nn.Layer):
def __init__(self): def __init__(self):
super(IdentityHead, self).__init__() super(IdentityHead, self).__init__()
def forward(self, x, label): def forward(self, x, label=None):
return {"features": x, "logits": None} return {"features": x, "logits": None}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册