丁云鹏 5 months ago
parent
commit
92c0f01f4a
1 changed files with 2 additions and 2 deletions
  1. 2 2
      recommend-model-produce/src/main/python/tools/utils/save_load.py

+ 2 - 2
recommend-model-produce/src/main/python/tools/utils/save_load.py

@@ -77,8 +77,8 @@ def save_inference_model(model_path,
     model_prefix = os.path.join(model_path, prefix)
 
 
-    program = paddle.static._get_valid_program(None)
-    program = paddle.static.normalize_program(
+    program = _get_valid_program(None)
+    program = normalize_program(
         program,
         feed_vars,
         fetch_vars,