|
@@ -17,12 +17,14 @@ runner:
|
|
|
train_reader_path: "bq_reader_train" # importlib format
|
|
train_reader_path: "bq_reader_train" # importlib format
|
|
|
train_batch_size: 8
|
|
train_batch_size: 8
|
|
|
model_save_path: "output_model_dssm"
|
|
model_save_path: "output_model_dssm"
|
|
|
|
|
+ split_file_list: true
|
|
|
|
|
|
|
|
reader_type: "QueueDataset" # DataLoader / QueueDataset / RecDataset
|
|
reader_type: "QueueDataset" # DataLoader / QueueDataset / RecDataset
|
|
|
pipe_command: "python bq_reader_train_ps.py"
|
|
pipe_command: "python bq_reader_train_ps.py"
|
|
|
thread_num: 1
|
|
thread_num: 1
|
|
|
sync_mode: "async"
|
|
sync_mode: "async"
|
|
|
|
|
|
|
|
|
|
+
|
|
|
use_gpu: False
|
|
use_gpu: False
|
|
|
epochs: 1
|
|
epochs: 1
|
|
|
print_interval: 1
|
|
print_interval: 1
|
丁云鹏