AttributeError:module ‘torch.optim‘ has no attribute ‘AdamW‘

最新推荐文章于 2023-05-26 21:22:14 发布

三个臭皮姜

最新推荐文章于 2023-05-26 21:22:14 发布

阅读量6.7k

点赞数 2

分类专栏： pytorch1.4.0

本文链接：https://blog.csdn.net/Aiden_yan/article/details/108563144

版权

4 篇文章 0 订阅

订阅专栏

在远程服务器上跑代码时遇到了这个问题：

AttributeError:module ‘torch.optim’ has no attribute ‘AdamW’

出错代码：

optimizer = torch.optim.AdamW(filter(lambda p: p.requires_grad,model.parameters()),betas=betas,lr=learning_rate,weight_decay=weight_decay)

出现这个问题是因为pytorch版本的不同。
代码是用pytorch1.3.0写的，但是之前因为服务器的cuda版本太低只能装pytorch1.0.0版本的。

于是试着把AdamW改成Adam，没想到就成功了。

改之后：

optimizer = torch.optim.Adam(filter(lambda p: p.requires_grad,model.parameters()),betas=betas,lr=learning_rate,weight_decay=weight_decay)

关注