1、python版本不同,3不能打开2,torch.load加载模型报错
2、多GPU,单GPU,训练的GPU号不一样都报错
3、ModuleNotFoundError: No module named ‘models’,这个报错不知道什么原因
嗯,在训练文件夹中同一个目录下,运行测试都没问题,可是我在另一个文件下,运行它用绝对路径,也不能运行报错,实在不行了
4、之前可以运行加载模型后来又不可以,pytorch ,版本不同也很有可能运行不了模型,
5、torch.save和torch.load 调用的应该是pickle库,用pickle.load()要求python能够找到类的定义。
不想保留源码的话,可以把模型转化成TorchScript之后保存。 所以保存完整的 model param , 的一个模型,需要调用的源码, torch.save和torch.load 调用的应该是pickle库,用pickle.load()要求python能够找到类的定义。
不想保留源码的话,可以把模型转化成TorchScript之后保存。