How to read .pt file

1

I would like to know the general way to read .pt file generated by pytorch, since I want to figure out how the data is organized and read the values so that I can make some modification on it and change the structure.
Could somebody tell me how to read that? Thank you so much.

3 Likes
2

You can take a look at this tutorial for more details on how saving and loading models in PyTorch are done.

The model’s state_dict is just a regular python dictionary once it is loaded. Maybe this thread can help you understand how to manipulate it, i.e., add/or change parameters, etc.

5 Likes
3

I still can not make sense of it

4

What do you mean? You can just use torch.load.

5

Thanks. I want to use it in Django

6

The problem is installing torch in a virtual environment keeps failing to I can’t use torch.load

IMG_20201019_122512_736.jpg
IMG_20201019_122512_736.jpg4128×1920 2.56 MB

7

Please don’t post pictures/screenshots. Post the actual code/error in text.

Seems like you just did pip install torch which won’t work well on Windows. Follow the guidelines here: https://pytorch.org/get-started/locally/

PS: you probably want to use a fully packaged 64-bit version of Python, which you do not seem to be using right now (wheel missing, 32-bit version). So e.g. this one.

8

Can models saved in .pt format be used to inference in python? I know that .pth files can be used, are .pt files used in the same way?

Thanks!