Once you have loaded the model (with net = torch.load(file)) or the model weights (with net.load_state_dict(torch.load(file))) correctly, then you can save the model simply by calling torch.save(net, path), as stated in the tutorial here 127.