monai
medical
lung_nodule_ct_detection / configs /inference_trt.json
katielink's picture
add the ONNX-TensorRT way of model conversion
967fa85
raw
history blame contribute delete
280 Bytes
{
"imports": [
"$import glob",
"$import os",
"$import torch_tensorrt"
],
"force_sliding_window": true,
"handlers#0#_disabled_": true,
"network_def": "$torch.jit.load(@bundle_root + '/models/model_trt.ts')",
"evaluator#amp": false
}