5. Production

TorchScript

Compiling your Python PyTorch code into a C++ binary layout for blazing speed.

python
traced_model = torch.jit.trace(model, dummy_input)
traced_model.save("model.pt")