cannot import name 'Nadam' from 'timm.optim.nadam

Question

E:\ProgramFiles>python main.py --model edgenext_small --eval True --batch_size 16 --data_path "E:\ProgramFiles\dataset\ImageNet - 1K" --output_dir E:\ProgramFiles\EdgeNeXt_results --resume edgenext_small.pth
Traceback (most recent call last):
  File "E:\ProgramFiles\main.py", line 16, in <module>
    from optim_factory import create_optimizer
  File "E:\ProgramFiles\optim_factory.py", line 7, in <module>
    from timm.optim.nadam import Nadam
ImportError: cannot import name 'Nadam' from 'timm.optim.nadam' (E:\hispark\python\Lib\site-packages\timm\optim\nadam.py)

从 timm.optim.nadam 导入 Nadam 失败，目前timm和PyTorch 都是最新版本，结果还一直像上图报错，要怎么解决？

Answer 1

出现 ImportError: cannot import name 'Nadam' from 'timm.optim.nadam' 错误，通常是因为 timm 库的版本或库内部结构与代码预期不一致导致无法正确导入 Nadam。以下是一些可以尝试的解决方法：

1. 检查 timm 版本：

   • 确认当前安装的 timm 版本是否包含 Nadam 优化器。可以查看 timm 官方文档（https://rwightman.github.io/pytorch-image-models/optimizers/），确认 Nadam 是否存在于当前版本中。
   • 如果当前版本不包含 Nadam，可以尝试升级或降级 timm 版本。例如，使用 pip 安装指定版本的 timm：

     # 升级到最新版本
     pip install --upgrade timm
     # 安装特定版本（例如 0.4.12 版本）
     pip install timm==0.4.12
     

2. 查看 timm.optim.nadam 模块内容：

   • 打开 E:\hispark\python\Lib\site-packages\timm\optim\nadam.py 文件，查看其中是否真的定义了 Nadam。
   • 如果 Nadam 确实存在于该模块中，但仍然无法导入，可能是因为模块加载顺序或命名空间问题。可以尝试使用绝对导入或相对导入的方式，确保导入路径正确。例如：

     from timm.optim import Nadam  # 尝试直接从 timm.optim 导入 Nadam
     

3. 确认是否有其他同名模块干扰：

   • 检查项目中是否存在其他名为 nadam.py 的文件，这可能会导致命名冲突。如果有，重命名或移除这些文件，确保 timm 库的 nadam 模块能够正确导入。

4. 检查 Python 环境：

   • 确保当前使用的 Python 环境是正确的，并且没有损坏。可以尝试在新的虚拟环境中重新安装 timm 库，然后运行代码。
   • 激活虚拟环境（如果使用虚拟环境），然后使用 pip 安装 timm：

     # 激活虚拟环境（以 Windows 下的虚拟环境为例）
     venv\Scripts\activate
     # 安装 timm
     pip install timm
     

通过以上步骤，应该能够解决 ImportError 问题，成功导入 Nadam 优化器。