diff --git a/monai/networks/utils.py b/monai/networks/utils.py index 7e3d2e2170..152911f443 100644 --- a/monai/networks/utils.py +++ b/monai/networks/utils.py @@ -986,6 +986,7 @@ def scale_batch_size(input_shape: Sequence[int], scale_num: int): inputs=input_placeholder, enabled_precisions=convert_precision, device=target_device, + ir="torchscript", **kwargs, )