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.