[54a97f]: / configs / inference_trt.json

Download this file

11 lines (10 with data), 246 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
{
"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
}