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




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.