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()
欢迎关注我的CSDN博客心系五道口,有问题请私信2395856915@qq.com