UserWarning: size_average and reduce args will be deprecated, please use reduction=‘sum‘ instead. wa

此报错为pytorch不同版本进行更新迭代时引起的警告，某些参数被取代了修正：criterion = torch.nn.BCELoss(size_average=True)改为：criterion = torch.nn.BCELoss(reduction='mean')criterion = torch.nn.BCELoss(size_average=False)改为：criterion = to

头发是我最后的倔强

12619人浏览 · 2022-03-28 20:26:24

头发是我最后的倔强  ·  2022-03-28 20:26:24 发布

此报错为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')


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

# pytorch # cnn # python

华为云开发者联盟

为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。

加入社区

更多推荐

• · 小窗口大魔力，实况窗服务实时掌控重要信息变化
• · GeminiDB全面联动MySQL：热点数据，一键加速
• · 基于Python和TensorFlow实现BERT模型应用

小窗口大魔力，实况窗服务实时掌控重要信息变化

华为云开发者联盟

GeminiDB全面联动MySQL：热点数据，一键加速

华为云开发者联盟

基于Python和TensorFlow实现BERT模型应用

华为云开发者联盟

欢迎加入社区

取消 确定