Torchscript and JIT

torch.jit.trace()         # takes your module or function and an example
                          # data input, and traces the computational steps
                          # that the data encounters as it progresses through the model
@script                   # decorator used to indicate data-dependent
                          # control flow within the code being traced
Comments