2020年2月25日
MSELoss(size_average=False)
摘要： nn.MSELoss()得到的是平均到像素的loss nn.MSELoss(size_average=False)得到整个batch所有像素loss和 MSELoss(size_average=False).div(batch_size)得到平均图像loss a=torch.Tensor([[1,1 阅读全文
posted @ 2020-02-25 16:44 Junzhao 阅读(1926) 评论(0) 推荐(0) 编辑