def pixel_wise_loss(self, x, y):
if self.loss_norm == 1:
return F.mean_absolute_error(x, y)
elif self.loss_norm == 2:
return F.mean_squared_error(x, y)
else:
raise ValueError('Invalid norm {}'.format(self.loss_norm))
评论列表
文章目录