ImportError: cannot import name 'AdamW' from 'transformers' 错误通常是由于 transformers 库版本过旧或导入路径变化引起

解决方法:

1.升级 transformers 库

AdamW 优化器在较新版本的 transformers 中可能需要通过 optimization 子模块导入,或者已被弃用。请先升级库到最新版本。

pip install --upgrade transformers

2. 检查导入方式

根据你的 transformers 版本,可能需要调整导入语句:

  • 如果版本 >= 4.0.0

    # 从 optimization 子模块导入
    from transformers.optimization import AdamW
  • 如果版本较旧
    建议升级到新版本。如果无法升级,可以直接使用 PyTorch 自带的 AdamW

  • from torch.optim import AdamW # PyTorch 1.6+ 已内置 AdamW

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐