fix export (#3393) (#3410)

PaddlePaddle · Feb 19, 2025 · 272c493 · 272c493
1 parent 99e301f
commit 272c493
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py b/paddlex/repo_apis/Paddle3D_api/bev_fusion/model.py
@@ -188,6 +188,17 @@ def export(self, weight_path, save_dir, **kwargs):
         if save_dir is not None:
             cli_args.append(CLIArgument("--save_dir", save_dir))
 
+        cli_args.append(CLIArgument("--save_name", "inference"))
+        cli_args.append(CLIArgument("--save_inference_yml"))
+
+        # PDX related settings
+        uniform_output_enabled = kwargs.pop("uniform_output_enabled", True)
+        export_with_pir = kwargs.pop("export_with_pir", False)
+        config.update({"uniform_output_enabled": uniform_output_enabled})
+        config.update({"pdx_model_name": self.name})
+        if export_with_pir:
+            config.update({"export_with_pir": export_with_pir})
+
         self._assert_empty_kwargs(kwargs)
         with self._create_new_config_file() as config_path:
             config.dump(config_path)