torch.jit.save()

Export

model_scripted = torch.jit.script(model)
model_scripted.save('model_scripted.pt')

Load

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