Skip to content

Commit 5a85a48

Browse files
authored
Merge branch 'main' into add-torchtext-tutorial
2 parents 525614c + f87faf7 commit 5a85a48

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

prototype_source/torchscript_freezing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
------------
1313
Model freezing can be invoked using API below:
1414
15-
``torch.jit.freeze(mod : ScriptModule, names : str[]) -> SciptModule``
15+
``torch.jit.freeze(mod : ScriptModule, names : str[]) -> ScriptModule``
1616
1717
Note the input module can either be the result of scripting or tracing.
1818
See https://pytorch.org/tutorials/beginner/Intro_to_TorchScript_tutorial.html

0 commit comments

Comments
 (0)