Can't saving model for inference using torch.save

PyTorch Live
1

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

model
model1459×902 60.5 KB

2

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.