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