PyTorch 2 · Performance
Speed up a model
Compile the model with torch.compile for faster training and inference.
model = torch.compile(model)
Official PyTorch 2 documentation
PyTorch 2 · Performance
Compile the model with torch.compile for faster training and inference.
model = torch.compile(model)
Official PyTorch 2 documentation
Was this page helpful?