Skip to content

Commit 1b904cd

Browse files
zhangyubo0722zhangyubo0722
andauthored
fix pdx_model_name (#3939)
Co-authored-by: zhangyubo0722 <[email protected]>
1 parent f90c6d2 commit 1b904cd

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def train(
137137
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
138138
export_with_pir = kwargs.pop("export_with_pir", False)
139139
config.update({"Global.uniform_output_enabled": uniform_output_enabled})
140-
config.update({"Global.pdx_model_name": self.name})
140+
config.update({"Global.model_name": self.name})
141141
if export_with_pir:
142142
config.update({"Global.export_with_pir": export_with_pir})
143143

@@ -274,7 +274,7 @@ def export(self, weight_path: str, save_dir: str, **kwargs) -> CompletedProcess:
274274
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
275275
export_with_pir = kwargs.pop("export_with_pir", False)
276276
config.update({"Global.uniform_output_enabled": uniform_output_enabled})
277-
config.update({"Global.pdx_model_name": self.name})
277+
config.update({"Global.model_name": self.name})
278278
if export_with_pir:
279279
config.update({"Global.export_with_pir": export_with_pir})
280280

paddlex/repo_apis/PaddleOCR_api/text_rec/model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -137,7 +137,7 @@ def train(
137137
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
138138
export_with_pir = kwargs.pop("export_with_pir", False)
139139
config.update({"Global.uniform_output_enabled": uniform_output_enabled})
140-
config.update({"Global.pdx_model_name": self.name})
140+
config.update({"Global.model_name": self.name})
141141
if export_with_pir:
142142
config.update({"Global.export_with_pir": export_with_pir})
143143

@@ -274,7 +274,7 @@ def export(self, weight_path: str, save_dir: str, **kwargs) -> CompletedProcess:
274274
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
275275
export_with_pir = kwargs.pop("export_with_pir", False)
276276
config.update({"Global.uniform_output_enabled": uniform_output_enabled})
277-
config.update({"Global.pdx_model_name": self.name})
277+
config.update({"Global.model_name": self.name})
278278
if export_with_pir:
279279
config.update({"Global.export_with_pir": export_with_pir})
280280

0 commit comments

Comments
 (0)