fix deeprec&deselectk

Author: wangzhen38

models/multitask/dselect_k/config.yaml

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 runner:
-  train_data_dir: "data/sample_data.pickle"
+  train_data_dir: "data"
   train_reader_path: "multiMNIST_reader" # importlib format
   use_gpu: False
   use_auc: True
@@ -22,7 +22,7 @@ runner:
   print_interval: 2
   #model_init_path: "output_model/0" # init model
   model_save_path: "output_model_dselect_k"
-  test_data_dir: "data/sample_data.pickle"
+  test_data_dir: "data"
   infer_batch_size: 8
   infer_reader_path: "multiMNIST_reader" # importlib format
   infer_load_path: "output_model_dselect_k"

models/rank/deeprec/trainer.py

@@ -18,16 +18,16 @@
 import logging
 import sys
 
-from tools.utils.utils_single import load_yaml, load_dy_model_class, \
-    get_abs_model, create_data_loader
-from tools.utils.save_load import load_model, save_model
-import argparse
-
 __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, create_data_loader
+from tools.utils.save_load import load_model, save_model
+import argparse
+
 logging.basicConfig(
     format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
 logger = logging.getLogger(__name__)