dssm train

recommend-model-produce/src/main/python/models/dssm/config_ps_hdfs.yaml

@@ -15,13 +15,13 @@
 runner:
   train_data_dir: "/dw/recommend/model/55_dssm_i2i_traindata/"
   train_reader_path: "bq_reader_train_ps"  # importlib format
-  train_batch_size: 64
+  train_batch_size: 1024
   model_save_path: "output_model_dssm"
   split_file_list: true
 
   reader_type: "QueueDataset"  # DataLoader / QueueDataset / RecDataset
   pipe_command: "python bq_reader_train_ps.py"
-  thread_num: 2
+  thread_num: 1
   sync_mode: "async"