import paddle
import numpy as np
class HeadLoss(paddle.nn.Layer):
def __init__(self):
super(HeadLoss, self).__init__()
import paddle
import numpy as np
class HeadLoss(paddle.nn.Layer):
def __init__(self):
super(HeadLoss, self).__init__()