运行报错
from timm.optim.novograd import NovoGrad
ModuleNotFoundError: No module named ‘timm.optim.novograd’。
问题原因
timm版本过高,novograd函数已被抛弃。
解决办法
方法1:安装更低版本的timm
pip install timm==0.4.12
方法2:使用NovoGrad优化函数 NvNovoGrad
全局搜索NovoGrad修改为NvNovoGrad,主要包括import和optimizer定义两部分:
from timm.optim.nvnovograd import NvNovoGrad
optimizer = NvNovoGrad(略)