Skip to content

Commit 706b7b7

Browse files
juncaipengwozna
andauthored
Fix problem with py3.6 and test for quant2_int8_lstm (#41420) (#42447)
Co-authored-by: joanna.wozna.intel <[email protected]>
1 parent b57c132 commit 706b7b7

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

python/paddle/distributed/parallel.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from paddle.distributed.fleet.launch_utils import check_backend
3131
from paddle.fluid.dygraph.parallel import ParallelEnv
3232
from paddle.distributed.fleet.base.private_helper_function import wait_server_ready # noqa: F401
33+
from paddle.distributed import collective
3334
from paddle.distributed.collective import _set_group_map
3435
from paddle.distributed.collective import _set_group_map_by_name
3536
from paddle.distributed.collective import _get_group_map_by_name

python/paddle/fluid/contrib/slim/tests/save_quant_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,8 @@ def transform_and_save_int8_model(original_path,
8888
debug=False,
8989
quant_model_filename='',
9090
quant_params_filename='',
91-
save_model_filename='',
92-
save_params_filename=''):
91+
save_model_filename="__model__",
92+
save_params_filename=None):
9393
place = fluid.CPUPlace()
9494
exe = fluid.Executor(place)
9595
inference_scope = fluid.executor.global_scope()

0 commit comments

Comments
 (0)