|
@@ -13,14 +13,14 @@
|
|
|
# limitations under the License.
|
|
# limitations under the License.
|
|
|
|
|
|
|
|
runner:
|
|
runner:
|
|
|
- train_data_dir: "/dypdemo"
|
|
|
|
|
- train_reader_path: "bq_reader_train" # importlib format
|
|
|
|
|
|
|
+ train_data_dir: "/dyp"
|
|
|
|
|
+ train_reader_path: "bq_reader_train_ps" # 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
|
|
split_file_list: true
|
|
|
|
|
|
|
|
reader_type: "QueueDataset" # DataLoader / QueueDataset / RecDataset
|
|
reader_type: "QueueDataset" # DataLoader / QueueDataset / RecDataset
|
|
|
- pipe_command: "python bq_reader_train.py"
|
|
|
|
|
|
|
+ pipe_command: "python bq_reader_train_ps.py"
|
|
|
thread_num: 1
|
|
thread_num: 1
|
|
|
sync_mode: "async"
|
|
sync_mode: "async"
|
|
|
|
|
|
|
@@ -30,7 +30,7 @@ runner:
|
|
|
print_interval: 1
|
|
print_interval: 1
|
|
|
|
|
|
|
|
test_data_dir: "data/test"
|
|
test_data_dir: "data/test"
|
|
|
- infer_reader_path: "bq_reader_infer" # importlib format
|
|
|
|
|
|
|
+ infer_reader_path: "bq_reader_train_ps" # importlib format
|
|
|
infer_batch_size: 1
|
|
infer_batch_size: 1
|
|
|
infer_load_path: "/app/output_model_dssm"
|
|
infer_load_path: "/app/output_model_dssm"
|
|
|
infer_start_epoch: 0
|
|
infer_start_epoch: 0
|
丁云鹏