infer

recommend-model-produce/src/main/python/tools/static_ps_infer_v2.py

@@ -15,7 +15,7 @@
 from __future__ import print_function
 import os
 os.environ['FLAGS_enable_pir_api'] = '0'
-from utils.static_ps.reader_helper import get_reader, get_example_num, get_file_list, get_word_num
+from utils.static_ps.reader_helper import get_infer_reader
 from utils.static_ps.program_helper import get_model, get_strategy, set_dump_config
 from utils.static_ps.metric_helper import set_zero, get_global_auc
 from utils.static_ps.common_ps import YamlHelper, is_distributed_env