Monai, Medical Imaging, Pytorch, Triton Inference Server, Torchserve.
def test_segresnet(batch_size=1): handler = ModelHandler() print(EXAMPLE_ROOT_DIR.as_posix()) ctx = MockContext(model_pt_file = MODEL_PT_FILE, model_dir= EXAMPLE_ROOT_DIR.as_posix(), model_file = None,) handler.initialize(ctx) handler.context = ctx handler.handle(TEST_DATA, ctx) if __name__ ==