def loss(output, target, *args):
assert isinstance(output, Variable) and isinstance(target, Variable)
# return torch.mean(torch.sum((output - target).clamp(-1, 1) ** 2, dim=1))
return F.smooth_l1_loss(output, target, size_average=False)
评论列表
文章目录