Monai, Medical Imaging, Pytorch, Triton Inference Server, Torchserve.
ModelDefinition(model_name='SegResNet') model = model_def.get_model() model_fn = './model/SegResNet/best_metric_model.pth' # Filename model.load_state_dict(torch.load(model_fn)) x = torch.zeros(1, 1, 256, 256, 24) traced_model = torch.jit.trace(model, x)