How can I load a trained model in python with torch?

Question

How can I load a trained model in python with torch?

sun-peach opened this issue 4 years ago · 2 comments

Hi, I have followed the librispeech cfg to train a model. Now I would like to load it in python code with torch. I will use it as a pretrained module within a complex system. Do you have any tutorial to do that? Thank you.

Answer 1 · 2020-07-23T14:54:33.000Z

The .pkl that is saved is pretty standard ! You can load it with torch.load()

Answer 2 · 2020-07-26T20:44:34.000Z

Thank you. Will it save the intermediate model somewhere? I make it run 15 epochs, while I don't see any .pkl files in the corresponding model directory (I set the out_folder in the config file) during training (currently, it has run 9 epochs). Should I just wait until it finish all the epochs?