|
|
@ -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)
|
|
|
|