pytorch保存整个模型时因为相对位置改变导致ModuleNotFoundError: No module named ‘networks‘

1.pytorch训练好的模型在测试时出现如下错误

2.除了网上说的改为只加载参数外,以下方法亲测有用,在主运行代码中加入以下三行代码即可

import os,sys

root_path = os.getcwd()

sys.path.insert(0,root_path+"/yolov5") #这个后面的"/yolov5" 文件夹应该是指向的 yolov5根目录.

 

你可能感兴趣的:(pytorch)