Can't saving model for inference using

So I can not saving model after training. Can anyone help me? Thank

I’m not familiar enough with torch2trt so would recommend to either use TorchTRT or create an issue in the torch2trt GitHub repository so that the code owners could take a look.