Skip to content

Commit 272c493

Browse files
authored
fix export (#3393) (#3410)
1 parent 99e301f commit 272c493

File tree

1 file changed

+11
-0
lines changed
  • paddlex/repo_apis/Paddle3D_api/bev_fusion

1 file changed

+11
-0
lines changed

paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -188,6 +188,17 @@ def export(self, weight_path, save_dir, **kwargs):
188188
if save_dir is not None:
189189
cli_args.append(CLIArgument("--save_dir", save_dir))
190190

191+
cli_args.append(CLIArgument("--save_name", "inference"))
192+
cli_args.append(CLIArgument("--save_inference_yml"))
193+
194+
# PDX related settings
195+
uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
196+
export_with_pir = kwargs.pop("export_with_pir", False)
197+
config.update({"uniform_output_enabled": uniform_output_enabled})
198+
config.update({"pdx_model_name": self.name})
199+
if export_with_pir:
200+
config.update({"export_with_pir": export_with_pir})
201+
191202
self._assert_empty_kwargs(kwargs)
192203
with self._create_new_config_file() as config_path:
193204
config.dump(config_path)

0 commit comments

Comments
 (0)