File size: 307 Bytes
fd4ffa6 |
1 2 3 4 5 6 7 8 9 10 11 |
{
"imports": [
"$import numpy",
"$from monai.networks import trt_compile"
],
"trt_args": {
"dynamic_batchsize": "$[1, @inferer#sw_batch_size, @inferer#sw_batch_size]"
},
"network": "$trt_compile(@network_def.to(@device), @pretrained_ckpt_path, args=@trt_args)"
}
|