|
|
@@ -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"
|
|
|
|
|
|
|
丁云鹏