Skip to content

Commit 6a130cc

Browse files
authored
7713 Update TRT parameter (#7714)
Fixes #7713 . ### Description A few sentences describing the changes proposed in this pull request. ### Types of changes <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [x] Non-breaking change (fix or new feature that would not break existing functionality). - [ ] Breaking change (fix or new feature that would cause existing functionality to change). - [ ] New tests added to cover the changes. - [x] Integration tests passed locally by running `./runtests.sh -f -u --net --coverage`. - [ ] Quick tests passed locally by running `./runtests.sh --quick --unittests --disttests`. - [ ] In-line docstrings updated. - [ ] Documentation updated, tested `make html` command in the `docs/` folder. Signed-off-by: binliu <binliu@nvidia.com>
1 parent 8c709de commit 6a130cc

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

monai/networks/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -986,6 +986,7 @@ def scale_batch_size(input_shape: Sequence[int], scale_num: int):
986986
inputs=input_placeholder,
987987
enabled_precisions=convert_precision,
988988
device=target_device,
989+
ir="torchscript",
989990
**kwargs,
990991
)
991992

0 commit comments

Comments
 (0)