mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
10 lines
209 B
Python
10 lines
209 B
Python
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}
|