Merge pull request #366 from seemingwang/master

decrease epoch number

Author: frankwhzhang

models/rank/naml/README.md

@@ -118,4 +118,4 @@ python3 -u ../../../tools/trainer.py -m config_bigdata.yaml
 python3 -u ../../../tools/infer.py -m config_bigdata.yaml
 ```
 
-期待预测auc为0.65
+期待预测auc为0.66

models/rank/naml/config_bigdata.yaml

@@ -17,16 +17,16 @@ runner:
   train_reader_path: "NAMLDataReader" # importlib format
   use_gpu: False
   train_batch_size: 50
-  epochs: 3
+  epochs: 2
   print_interval: 10
   #model_init_path: "output_model/0" # init model
   model_save_path: "output_model_all"
   infer_batch_size: 10
   infer_reader_path: "NAMLDataReader" # importlib format
   test_data_dir:  "../../../datasets/MIND/data/test"
   infer_load_path: "output_model_all"
-  infer_start_epoch: 3
-  infer_end_epoch: 4
+  infer_start_epoch: 1
+  infer_end_epoch: 2
 
 # hyper parameters of user-defined network
 hyper_parameters: