configs/inference_trt.yaml

---
imports:
- "$import glob"
- "$import os"
- "$import torch_tensorrt"
network_def: "$torch.jit.load(@bundle_root + '/models/model_trt.ts')"
evaluator#amp: false
initialize:
- "$setattr(torch.backends.cudnn, 'benchmark', True)"