1.保存模型与加载
简单的保存与加载方法

#保存整个网络
torch.save(model,path)
#保存网络中的参数,速度快,占空间少
torch.save(model.state_dict(),path)
#-------------------------------------------
#加载的方法分别为:
torch.load(path)
model.load_state_dict(torch.load(path))

如果加载出现错误:

TypeError: 'IncompatibleKeys' object is not callable

一般来说torch加载的时候出现问题,主要是在model.load_state_dict()的时候,使用了model = model.load_state_dict() 只需要将model = 祛除即可。