【Debug】UserWarning: size_average and reduce args will be deprecated, please use reduction=‘sum‘

问题描述:

【Debug】UserWarning: size_average and reduce args will be deprecated, please use reduction=‘sum‘_第1张图片

报错原因:

  • 由于版本不同,PyTorch中的损失函数的参数,发生了变化

解决方式:

  • criterion = torch.nn.MSELoss(size_average=False)改为criterion = torch.nn.MSELoss(reduction='sum')

你可能感兴趣的:(Debug,Python/Pycharm,深度学习/PyTorch,pytorch,深度学习,机器学习)