qq_41645987的博客

class Center_loss(nn.Module): def __init__(self,src_class): super(Center_loss, self).__init__() self.n_class=src_class self.MSELoss = nn.MSELoss() # (x-y)^2 self.MSELoss = self.MSELoss.cuda() def forward(self,