模型的保存与加载

保存与加载整个模型的结构信息和参数信息

torch.save(model,'./model.pth')
load_model=torch.load('model.pth')

保存与加载整个模型的参数信息

torch.save(model.state_dict(),'./model_state.pth')
load_model=model.load_state_dic(torch.load('model_state.pth'))

load_model加载前需要先实例化