l2
def l2_loss(gt, pred):
B, C, H, W = gt.size()
# loss = t.sum(t.abs(gt - pred))
loss = t.sum((gt - pred) * (gt - pred)) / (B * C * H * W * 10)
return lossloss
猜你喜欢
转载自blog.csdn.net/a362682954/article/details/83346030
今日推荐
周排行