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

问题描述:

在这里插入图片描述

报错原因:

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

解决方式:

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

猜你喜欢

转载自blog.csdn.net/weixin_42306148/article/details/127464189