Is there a way to convert a local PyTorch model to Torch Scipt Model?

Mobile
1

Hi, guys, I’m learning PyTorch and a total beginner.
How do I convert a downloaded PyTorch model (https://drive.google.com/file/d/1K_xN32uoQKI8XmNYNLTX5gDn1UnQVe5I/view) to Torch Scirpt Model. I have tried to use Python packages, torchvision and torch, but I have no idea how to load a model form local path.
Any help would be appreciated.

2

I don’t know what’s exactly stored in this file, but the common workflow would be to create a model instance and load its state_dict as seen in this code snippet:

model = MyModel()
model.load_state_dict(torch.load(PATH_TO_STATE_DICT))
3

The code works fine. Thank you very much