Save and Load

1. Just for parameters:

# Save
torch.save(model.state_dict(), '\parameter.pkl')
# Load
model = TheModelClass(...)
model.load_state_dict(torch.load('\parameter.pkl'))

2. For entire model: 

# Save
torch.save(model, '\model.pkl')
# Load
model = torch.load('\model.pkl')