丁云鹏 5 ماه پیش
والد
کامیت
288321bd88
1فایلهای تغییر یافته به همراه1 افزوده شده و 0 حذف شده
  1. 1 0
      recommend-model-produce/src/main/python/tools/utils/save_load.py

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

@@ -16,6 +16,7 @@ import paddle
 import os
 import logging
 import numpy as np
+from paddle.static.io import _get_valid_program, normalize_program
 
 logging.basicConfig(
     format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)