Преглед на файлове

fix pdx_model_name (#3939)

Co-authored-by: zhangyubo0722 <zangyubo0722@163.com>
zhangyubo0722 преди 6 месеца
родител
ревизия
1b904cdda1
променени са 2 файла, в които са добавени 4 реда и са изтрити 4 реда
  1. 2 2
      paddlex/repo_apis/PaddleOCR_api/formula_rec/model.py
  2. 2 2
      paddlex/repo_apis/PaddleOCR_api/text_rec/model.py

+ 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})