We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
torchscript_freezing.py
SciptModule
ScriptModule
1 parent be6e863 commit f87faf7Copy full SHA for f87faf7
prototype_source/torchscript_freezing.py
@@ -12,7 +12,7 @@
12
------------
13
Model freezing can be invoked using API below:
14
15
- ``torch.jit.freeze(mod : ScriptModule, names : str[]) -> SciptModule``
+ ``torch.jit.freeze(mod : ScriptModule, names : str[]) -> ScriptModule``
16
17
Note the input module can either be the result of scripting or tracing.
18
See https://pytorch.org/tutorials/beginner/Intro_to_TorchScript_tutorial.html
0 commit comments