未验证 提交 5e09d67a 编写于 作者: S Shibo Tao 提交者: GitHub

fix API: normalize_program. test=develop (#33408)

* fix: paddle.static.default_main_program. test=develop

* add normalize_program to __all__. test=develop
上级 0549d4af
...@@ -85,6 +85,7 @@ __all__ = [ #noqa ...@@ -85,6 +85,7 @@ __all__ = [ #noqa
'load', 'load',
'save_inference_model', 'save_inference_model',
'load_inference_model', 'load_inference_model',
'normalize_program',
'load_program_state', 'load_program_state',
'set_program_state', 'set_program_state',
'cpu_places', 'cpu_places',
......
...@@ -157,7 +157,7 @@ def normalize_program(program, feed_vars, fetch_vars): ...@@ -157,7 +157,7 @@ def normalize_program(program, feed_vars, fetch_vars):
exe.run(paddle.static.default_startup_program()) exe.run(paddle.static.default_startup_program())
# normalize main program. # normalize main program.
program = default_main_program() program = paddle.static.default_main_program()
normalized_program = paddle.static.normalize_program(program, [image], [predict]) normalized_program = paddle.static.normalize_program(program, [image], [predict])
""" """
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册