Merge pull request #776 from wangzhen38/qa_ci

fix reader and config

Author: frankwhzhang

models/rank/deeprec/infer.py

@@ -19,18 +19,18 @@
 import sys
 from math import sqrt
 
+__dir__ = os.path.dirname(os.path.abspath(__file__))
+print(os.path.abspath('/'.join(__dir__.split('/')[:-3])))
+sys.path.append(os.path.abspath(os.path.join(__dir__, '..')))
+sys.path.append(os.path.abspath('/'.join(__dir__.split('/')[:-3])))
+
 from tools.utils.utils_single import load_yaml, load_dy_model_class, \
     get_abs_model
 from tools.utils.save_load import load_model
 from paddle.io import DataLoader
 import argparse
 from importlib import import_module
 
-__dir__ = os.path.dirname(os.path.abspath(__file__))
-print(os.path.abspath('/'.join(__dir__.split('/')[:-3])))
-sys.path.append(os.path.abspath(os.path.join(__dir__, '..')))
-sys.path.append(os.path.abspath('/'.join(__dir__.split('/')[:-3])))
-
 logging.basicConfig(
     format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
 logger = logging.getLogger(__name__)

models/rank/fat_deepffm/config.yaml

@@ -26,7 +26,7 @@ runner:
   print_interval: 10
 
   model_save_path: "output_model_fat_deepffm"
-  infer_batch_size: 1000
+  infer_batch_size: 1
   infer_reader_path: "criteo_reader" # importlib format
   test_data_dir: "data/sample_data/train"