@@ -17,12 +17,14 @@ runner:
train_reader_path: "bq_reader_train" # importlib format
train_batch_size: 8
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: 1
sync_mode: "async"
+
use_gpu: False
epochs: 1
print_interval: 1