torch.load() 出现 No module named ‘models‘

原来训练模型的服务器满了，我将模型移动到另一台服务器进行测试，由于测试过程中原来训练的有些文件夹未使用，因此我没有全部拷贝，只拷贝需要用的部分，然后就出现了如下错误：

 后来发现一种比较合理的解释是我们在训练train.py完后是这样保存权重的。torch.save(model, PATH)：

 

这样的方式保存的权重包，会将你模型的源代码的相对位置也默认保存如下这个默认路径，所以使用torch.load() 的时候也要保持这个路径，即utis,HDF5，datasets等这些文件夹也必须带上。