Search code examples
pythonmachine-learningpytorchtorch

how to convert .pth.tar file to a .pt?

I have a .pth.tar file how can I read in that file and then save it to the same directory in a .pt file?

I am using the modelzoo mlnf file which I cannot find its .pt equivalent.

https://kaiyangzhou.github.io/deep-person-reid/MODEL_ZOO.html

Solution

  • It turns out you can load your .pth.tar file directly with torch.load:

    state_dict = torch.load(model_name)

    See this thread for reference.