fix bugs from bert4rec+ensfm

Author: wangzhen38

datasets/ml-1m_ensfm/run.sh

@@ -13,7 +13,7 @@
 # limitations under the License.
 #!/bin/bash
 
-
+mkdir data
 wget https://paddlerec.bj.bcebos.com/recall/ensfm/ml-1m-ensfm.tar.gz
 tar zxvf ml-1m-ensfm.tar.gz -C data
 rm ml-1m-ensfm.tar.gz

models/rank/bert4rec/config_bigdata.yaml

@@ -20,16 +20,16 @@ runner:
   use_gpu: True
   train_batch_size: 1
   data_batch_size: 256
-  epochs: 10
+  epochs: 50
   print_interval: 100
 
   model_save_path: "output_model_bert4rec_all"
   test_data_dir: "../../../datasets/Beauty/data/test"
   infer_reader_path: "data_reader" # importlib format
   infer_batch_size: 1
   infer_load_path: "output_model_bert4rec_all"
-  infer_start_epoch: 9
-  infer_end_epoch: 10
+  infer_start_epoch: 49
+  infer_end_epoch: 50
 
 
 # hyper parameters of user-defined network