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


错误提示:

 

 为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()

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM