identity_head.py 209 B

123456789
  1. from paddle import nn
  2. class IdentityHead(nn.Layer):
  3. def __init__(self):
  4. super(IdentityHead, self).__init__()
  5. def forward(self, x, label=None):
  6. return {"features": x, "logits": None}