未验证 提交 42aefbb4 编写于 作者: W Walter 提交者: GitHub

Merge pull request #795 from weisy11/develop_reg

fix bug
from paddle import nn
class IdentityHead(nn.Layer):
def __init__(self):
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}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册