Save and Load

# Save

torch.save(model.state_dict(), '\parameter.pkl')

# Load

model = TheModelClass(...)

model.load_state_dict(torch.load('\parameter.pkl'))

# Save

torch.save(model, '\model.pkl')

# Load

model = torch.load('\model.pkl')