mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
6 lines
184 B
Python
6 lines
184 B
Python
class IdentityHead(nn.Layer):
|
|
def __init__(self):
|
|
super(IdentityHead, self).__init__()
|
|
|
|
def forward(self, x, label=None):
|
|
return {"features": x, "logits": None} |