2021-06-07 15:05:26 +08:00
|
|
|
from paddle import nn
|
|
|
|
|
|
|
|
|
|
|
2021-06-07 14:22:53 +08:00
|
|
|
class IdentityHead(nn.Layer):
|
|
|
|
|
def __init__(self):
|
|
|
|
|
super(IdentityHead, self).__init__()
|
|
|
|
|
|
|
|
|
|
def forward(self, x, label=None):
|
2021-06-07 15:05:51 +08:00
|
|
|
return {"features": x, "logits": None}
|
