|
|
@@ -0,0 +1,47 @@
|
|
|
+# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
|
|
|
+#
|
|
|
+# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
+# you may not use this file except in compliance with the License.
|
|
|
+# You may obtain a copy of the License at
|
|
|
+#
|
|
|
+# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
+#
|
|
|
+# Unless required by applicable law or agreed to in writing, software
|
|
|
+# distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
+# See the License for the specific language governing permissions and
|
|
|
+# limitations under the License.
|
|
|
+
|
|
|
+runner:
|
|
|
+ train_data_dir: "data/train"
|
|
|
+ train_reader_path: "bq_reader_train" # importlib format
|
|
|
+ train_batch_size: 8
|
|
|
+ model_save_path: "output_model_dssm_debug"
|
|
|
+
|
|
|
+ reader_type: "QueueDataset" # DataLoader / QueueDataset / RecDataset
|
|
|
+ pipe_command: "python bq_reader_train_ps.py"
|
|
|
+ thread_num: 1
|
|
|
+ sync_mode: "async"
|
|
|
+
|
|
|
+ use_gpu: False
|
|
|
+ epochs: 5
|
|
|
+ print_interval: 1
|
|
|
+
|
|
|
+ test_data_dir: "data/test"
|
|
|
+ infer_reader_path: "bq_reader_infer" # importlib format
|
|
|
+ infer_batch_size: 1
|
|
|
+ infer_load_path: "output_model_dssm"
|
|
|
+ infer_start_epoch: 0
|
|
|
+ infer_end_epoch: 1
|
|
|
+ infer_reader_type: "QueueDataset"
|
|
|
+hyper_parameters:
|
|
|
+ optimizer:
|
|
|
+ class: adam
|
|
|
+ learning_rate: 0.001
|
|
|
+ strategy: sync
|
|
|
+ trigram_d: 2900
|
|
|
+ neg_num: 1
|
|
|
+ is_infer: True
|
|
|
+ slice_end: 8
|
|
|
+ fc_sizes: [300, 300, 128]
|
|
|
+ fc_acts: ['relu', 'relu', 'relu']
|
often