I have trained a model in python with pytorch and have saved it like this:
traced_script_module = torch.jit.script(net) traced_script_module.save("net_0.pt&q