UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead. warnings.warn(warning.format(ret))

时间：2021-08-24 16:46:36 阅读：19 评论：0 收藏：0 [点我收藏+]

错误提示：

技术分享图片

为pytorch不同版本进行更新迭代时引起的警告，某些参数被取代了

修正：

criterion = torch.nn.BCELoss(size_average=True)
改为：
criterion = torch.nn.BCELoss(reduction=‘mean‘)

criterion = torch.nn.BCELoss(size_average=False)
改为：
criterion = torch.nn.BCELoss(reduction=‘sum‘)

其它损失函数更改方法类似

原代码：

test_loss += F.nll_loss(output, target, size_average=False).item()

修正代码：

test_loss += F.nll_loss(output, target, reduction=‘sum‘).item()

原文：https://www.cnblogs.com/pistachio0812/p/15180720.html

踩

(0)

评论一句话评论（0）

分享档案

更多>