Analyzes the entire model code to generate a TorchScript model which contains dynamic path
torch.jit
import torch import torch.nn as nn class Model(nn.Module): def __init__(self): super().__init__() self.fc = nn.Linear(10, 1) def forward(self, x): return self.fc(x) model = Model() scripted_model = torch.jit.script(model) scripted_model.save("model.pt")