Monai, Medical Imaging, Pytorch, Triton Inference Server, Torchserve.
traced_model.save('./model/SegResNet/traced_segres_model.pt') This model is saved in a directory. Now, we need to write the config.pbtxt file. name: "lv_segmentation" platform: "pytorch_libtorch" max_batch_size: 2 input [ { name: "input__0" data_type: TYPE_FP32 dims: [1, 256, 256, 24 ] } ] output [