PyTorch How to save a pretrained model like efficientnet_b0 model locally and load it?

[emadyounan]

Good evening everyone,

I'm trying to use a pre-trained model like efficientnet_b0 But I'm confused about how to save it on my PC and how to load it again on the client's PC without the internet!!

I used:

model_scripted = torch.jit.script(model) # Export to TorchScript
model_scripted.save('model_scripted.pt') # Save

and the load

model_0 = torch.jit.load('model_scripted.pt')
model_0.eval()

It works well but is that the right way to do it? or is there another way?

Thanks for help