fix bug in fleet, test=develop (#18058)

revert-18229-add_multi_gpu_install_check
tangwei12 6 years ago committed by GitHub
parent 42f12a4aca
commit 4c735f24ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -144,16 +144,9 @@ class DistributedTranspiler(Fleet):
executor, main_program, None, None, executor, main_program, None, None,
export_for_deployment) export_for_deployment)
else: else:
io.save_inference_model( io.save_inference_model(dirname, feeded_var_names, target_vars,
dirname, executor, self._origin_program, None, None,
feeded_var_names, export_for_deployment, True)
target_vars,
executor,
self._origin_program,
None,
None,
export_for_deployment,
model_only=True)
model_basename = "__model__" model_basename = "__model__"
model_filename = os.path.join(dirname, model_basename) model_filename = os.path.join(dirname, model_basename)

Loading…
Cancel
Save