運行報錯
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(略)