Explorar el Código

fix pdx_model_name (#3939)

Co-authored-by: zhangyubo0722 <zangyubo0722@163.com>
zhangyubo0722 hace 6 meses
padre
commit
1b904cdda1

+ 2 - 2
paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py

@@ -137,7 +137,7 @@ class FormulaRecModel(BaseModel):
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
         export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
-        config.update({"Global.pdx_model_name": self.name})
+        config.update({"Global.model_name": self.name})
         if export_with_pir:
             config.update({"Global.export_with_pir": export_with_pir})
 
@@ -274,7 +274,7 @@ class FormulaRecModel(BaseModel):
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
         export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
-        config.update({"Global.pdx_model_name": self.name})
+        config.update({"Global.model_name": self.name})
         if export_with_pir:
             config.update({"Global.export_with_pir": export_with_pir})
 

+ 2 - 2
paddlex/repo_apis/PaddleOCR_api/text_rec/model.py

@@ -137,7 +137,7 @@ class TextRecModel(BaseModel):
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
         export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
-        config.update({"Global.pdx_model_name": self.name})
+        config.update({"Global.model_name": self.name})
         if export_with_pir:
             config.update({"Global.export_with_pir": export_with_pir})
 
@@ -274,7 +274,7 @@ class TextRecModel(BaseModel):
         uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
         export_with_pir = kwargs.pop("export_with_pir", False)
         config.update({"Global.uniform_output_enabled": uniform_output_enabled})
-        config.update({"Global.pdx_model_name": self.name})
+        config.update({"Global.model_name": self.name})
         if export_with_pir:
             config.update({"Global.export_with_pir": export_with_pir})