AttributeError: module ‘torch.nn‘ has no attribute ‘HuberLoss‘

在跟李沐大神的深度学习课程,在实现一个最基本的线性回归算法这章,尝试使用HuberLoss代替MSELoss时,出现错误:

AttributeError: module 'torch.nn' has no attribute 'HuberLoss'

查了官方文档,应该是torch版本问题。运行conda list,我装的torch版本是1.8.1,只有1.9.0以上的版本才有HuberLoss。运行pip install --upgrade torch更新torch版本,问题解决。

你可能感兴趣的:(别催了别催了小白在学习了,深度学习,pytorch,神经网络)