torch.jit.trace()

Creator
Creator
Seonglae ChoSeonglae Cho
Created
Created
2025 Apr 16 12:57
Editor
Edited
Edited
2025 Jun 16 21:0
Refs
Change
torch.nn
model to TorchScript model
Unlike
torch.jit.script()
, there may be errors in dynamic paths when loading saved models

Dynamic Shape Inference with TorchScript Shape Caching

 
 
 
 
 

Recommendations