保存&加载模型参数
# save parameter
if accuracy > best_accuracy:
torch.save(model.state_dict(), '.\\parameter.pk1') #save parameter
# load parameter
model = TheModelClass(...)
model.load_state_dict(torch.load('\\parameter.pkl')) #load parameter
保存&加载模型参数模型
torch.save(model, '.\\model.pkl') #save model
torch.load(model, '\\model.pkl') #load model