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 = 祛除即可。