Browse Source

using deployment format when saving model

often 5 months ago
parent
commit
c7ae4937f8

+ 2 - 1
recommend-model-produce/src/main/python/tools/static_ps_trainer.py

@@ -225,7 +225,8 @@ class Main(object):
                 fleet.save_inference_model(
                     self.exe, model_dir,
                     [feed.name for feed in self.inference_feed_var],
-                    self.inference_target_var)
+                    self.inference_target_varm
+                    export_for_deployment=True )
             else:
                 paddle.static.save_inference_model(
                     model_dir,