Hi, you need to simply instantiate the model and and load the state dict. Something like
model = Model() # create your model
model.load_state_dict(torch.load('model.pth')) #assuming you have the model weights saved in model.pth