# An instance of your model.
model = torch.load('my_model.pth')
model.eval()
# An example input you would normally provide to your model's forward() method.
example = torch.rand(1, 3, 224, 224)
 
# Use torch.jit.trace to generate a torch.jit.ScriptModule via tracing.
traced_script_module = torch.jit.trace(model.cuda(), example.cuda())

# save it
traced_script_module.save("model_cpp.pt")