丁云鹏 5 месяцев назад
Родитель
Сommit
92c0f01f4a
1 измененных файлов с 2 добавлено и 2 удалено
  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,