未验证 提交 706b7b7f 编写于 作者: C cc 提交者: GitHub

Fix problem with py3.6 and test for quant2_int8_lstm (#41420) (#42447)

Co-authored-by: Njoanna.wozna.intel <joanna.wozna@intel.com>
上级 b57c132a
...@@ -30,6 +30,7 @@ from paddle.fluid.dygraph import parallel_helper ...@@ -30,6 +30,7 @@ from paddle.fluid.dygraph import parallel_helper
from paddle.distributed.fleet.launch_utils import check_backend from paddle.distributed.fleet.launch_utils import check_backend
from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.dygraph.parallel import ParallelEnv
from paddle.distributed.fleet.base.private_helper_function import wait_server_ready # noqa: F401 from paddle.distributed.fleet.base.private_helper_function import wait_server_ready # noqa: F401
from paddle.distributed import collective
from paddle.distributed.collective import _set_group_map from paddle.distributed.collective import _set_group_map
from paddle.distributed.collective import _set_group_map_by_name from paddle.distributed.collective import _set_group_map_by_name
from paddle.distributed.collective import _get_group_map_by_name from paddle.distributed.collective import _get_group_map_by_name
......
...@@ -88,8 +88,8 @@ def transform_and_save_int8_model(original_path, ...@@ -88,8 +88,8 @@ def transform_and_save_int8_model(original_path,
debug=False, debug=False,
quant_model_filename='', quant_model_filename='',
quant_params_filename='', quant_params_filename='',
save_model_filename='', save_model_filename="__model__",
save_params_filename=''): save_params_filename=None):
place = fluid.CPUPlace() place = fluid.CPUPlace()
exe = fluid.Executor(place) exe = fluid.Executor(place)
inference_scope = fluid.executor.global_scope() inference_scope = fluid.executor.global_scope()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册