Skip to content

Commit 417efd2

Browse files
author
Wei
authored
Update getting_started_with_fx_path.rst (#1343)
1 parent f0735f9 commit 417efd2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

docsrc/tutorials/getting_started_with_fx_path.rst

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,8 @@ wrapper API that consists of the major steps needed to finish this converison. P
2727
def compile(
2828
module: nn.Module,
2929
input,
30-
max_batch_size: int = 2048,
31-
max_workspace_size=1 << 25,
30+
max_batch_size=2048,
31+
max_workspace_size=33554432,
3232
explicit_batch_dimension=False,
3333
lower_precision=LowerPrecision.FP16,
3434
verbose_log=False,
@@ -37,6 +37,7 @@ wrapper API that consists of the major steps needed to finish this converison. P
3737
cuda_graph_batch_size=-1,
3838
dynamic_batch=True,
3939
) -> nn.Module:
40+
4041
"""
4142
Takes in original module, input and lowering setting, run lowering workflow to turn module
4243
into lowered module, or so called TRTModule.

0 commit comments

Comments
 (0)