提交 c8817f94 编写于 作者: Michael Suo's avatar Michael Suo 提交者: Facebook Github Bot

fix default value for script

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/23542

Test Plan: Imported from OSS

Differential Revision: D16557122

Pulled By: suo

fbshipit-source-id: c86578aa2c55f44ed5d573d33874a82244df3d09
上级 6314af6e
......@@ -1059,7 +1059,7 @@ def _compile_and_register_class(obj, rcb, qualified_name):
_add_script_class(obj, qualified_name)
def script(obj, optimize=True, _frames_up=0, _rcb=None):
def script(obj, optimize=None, _frames_up=0, _rcb=None):
r"""
Scripting a function or ``nn.Module`` will inspect the source code, compile
it as TorchScript code using the TorchScript compiler, and return a ``ScriptModule`` or
......
Markdown 格式
0% or
您添加了 0 到此讨论。请谨慎行事。
先完成此消息的编辑!
想要评论请 注册