monai
medical
katielink's picture
add support for TensorRT conversion and inference
67d834c
raw
history blame contribute delete
191 Bytes
---
imports:
- "$import glob"
- "$import os"
- "$import torch_tensorrt"
handlers#0#_disabled_: true
network_def: "$torch.jit.load(@bundle_root + '/models/model_trt.ts')"
evaluator#amp: false