configs/inference_trt.json

{
    "+imports": [
        "$from monai.networks import trt_compile"
    ],
    "network": "$trt_compile(@network_def.to(@device), @checkpoint)"
}