Skip to content

Commit 564c40a

Browse files
committed
update hpi config
1 parent 58a6f48 commit 564c40a

File tree

141 files changed

+251
-4548
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

141 files changed

+251
-4548
lines changed

paddlex/inference/components/paddle_predictor/predictor.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,6 @@ def apply(self, **kwargs):
174174
for idx in range(len(x)):
175175
self.input_handlers[idx].reshape(x[idx].shape)
176176
self.input_handlers[idx].copy_from_cpu(x[idx])
177-
178177
self.predictor.run()
179178
output = []
180179
for out_tensor in self.output_handlers:

paddlex/repo_apis/PaddleClas_api/cls/model.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,6 @@ def train(
117117
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
118118
config.update([f"Global.uniform_output_enabled={uniform_output_enabled}"])
119119
config.update([f"Global.pdx_model_name={self.name}"])
120-
hpi_config_path = self.model_info.get("hpi_config_path", None)
121-
config.update([f"Global.hpi_config_path={hpi_config_path}"])
122120

123121
config.dump(config_path)
124122
self._assert_empty_kwargs(kwargs)
@@ -232,9 +230,9 @@ def export(self, weight_path: str, save_dir: str, **kwargs) -> CompletedProcess:
232230
if device:
233231
config.update_device(device)
234232
# PDX related settings
233+
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
234+
config.update([f"Global.uniform_output_enabled={uniform_output_enabled}"])
235235
config.update([f"Global.pdx_model_name={self.name}"])
236-
hpi_config_path = self.model_info.get("hpi_config_path", None)
237-
config.update([f"Global.hpi_config_path={hpi_config_path}"])
238236

239237
config.dump(config_path)
240238

0 commit comments

Comments
 (0)