Change torch.nn model to TorchScript modelUnlike torch.jit.script(), there may be errors in dynamic paths when loading saved modelsDynamic Shape Inference with TorchScript Shape Caching