Unable to save the model in TorchScript format?

I’m trying to save my model in TorchScript format, but unfortunately getting errors.

What you have already tried

torch.jit.script(model)

Environment

python

Build information about Torch-TensorRT can be found by turning on debug messages

  • PyTorch Version (e.g., 1.0):1.11.0+cu113
  • CPU Architecture:
  • OS (e.g., Linux): ubuntu 20.04
  • How you installed PyTorch (conda, pip, libtorch, source): pip
  • Build command you used (if compiling from source):
  • Are you using local sources or building from archives:
  • Python version:3.9.7
  • CUDA version:11.7
  • GPU models and configuration: RTX GEFORCE 2060
  • Any other relevant information:

To Reproduce

please find the original git repo. Here, I’m using ade20k-hrnetv2.yaml config file.

@ptrblck