加载权重BUG:在使用pytorch加载模型遇到问题ModuleNotFoundError: No module named ‘‘XXX“

项目场景：

使用torch.load(path, map_location=“cpu”)来加载模型

问题描述：

在使用pytorch加载模型遇到问题ModuleNotFoundError: No module named ‘‘XXX"

原因分析：

模型中使用了自定义的类‘‘Exploring"，加载的时候程序需要找到Exploring类

---

解决方案：

将需要的文件也加到工程里就可以了

import sys  
sys.path.append(r"D:\gw\mae-main\Exploring")
import Exploring

注意: 只要将需要的本地文件导入就可以成功加载权重，我这里用sys库是因为Exploring文件夹在我程序文件的上级目录里