提交 3b28af95 编写于 作者: M Megvii Engine Team

fix(sdk): add disable-optimize-for-inference option

GitOrigin-RevId: 7e6ea13be7eaa70cd0d49f20571c4014d3a20e07
上级 938152af
......@@ -17,11 +17,11 @@ import numpy as np
import megengine as mge
import megengine.core._imperative_rt as rt
import megengine.core.tensor.megbrain_graph as G
from megengine.utils import comp_graph_tools as cgtools
from megengine.core.ops import builtin
from megengine import tensor
from megengine.core._imperative_rt.core2 import apply
from megengine.core.ops import builtin
from megengine.core.tensor.megbrain_graph import VarNode
from megengine import tensor
from megengine.utils import comp_graph_tools as cgtools
logger = mge.get_logger(__name__)
......@@ -372,8 +372,8 @@ def main():
)
parser.add_argument(
"--optimize-for-inference",
action="store_false",
help="enbale optimization for inference",
action="store_true",
help="enable optimization for inference",
)
parser.add_argument(
"--no-assert",
......@@ -498,7 +498,7 @@ def main():
wrap_output_vars,
append_json=True,
strip_info_file=strip_info_file,
**sereg_kwargs
**sereg_kwargs,
)
fout.write(dump_content)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册