pytorch 保存训练好的模型

torch.save(model_object, 'model.pth') 
model = torch.load('model.pth')

torch.save(model_obj.state_dict(), 'params.pth')  
model_obj.load_state_dict(torch.load('params.pth'))  

torch.save({'state_dict': net.state_dict(), 'linear1':net.linear1.state_dict(),
            'optimizer': optimizer.state_dict(),'num_epoch':num_epochs },
            'detail.pth')

model = torch.load('detail.pth')
net = DNN(num_input,num_hidden1,num_hidden2,num_output)
net.load_state_dict(model['state_dict'])