From ac45ce075a35bce5fd1668adf68d8d3192f6b9d1 Mon Sep 17 00:00:00 2001
From: tangwei <tangwei12@baidu.com>
Date: Tue, 14 Apr 2020 14:24:38 +0800
Subject: [PATCH] user define trainer

---
 .../examples/ctr-dnn_train_userdefine.yaml    | 63 +++++++++++++++++++
 1 file changed, 63 insertions(+)
 create mode 100644 fleetrec/examples/ctr-dnn_train_userdefine.yaml

diff --git a/fleetrec/examples/ctr-dnn_train_userdefine.yaml b/fleetrec/examples/ctr-dnn_train_userdefine.yaml
new file mode 100644
index 00000000..2beec8d4
--- /dev/null
+++ b/fleetrec/examples/ctr-dnn_train_userdefine.yaml
@@ -0,0 +1,63 @@
+# 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.# Copyright (c) 2019 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.
+
+train:
+  threads: 12
+  epochs: 10
+  trainer:  "UserDefineTrainer"
+  location: "/root/FleetRec/fleetrec/examples/user_define_trainer.py"
+
+  reader:
+    batch_size: 2
+    class: "fleetrec.models.ctr_dnn.data_generator"
+    train_data_path: "/root/FleetRec/fleetrec/models/ctr_dnn/data/train/"
+
+  model:
+    models: "fleetrec.models.ctr_dnn.model"
+    hyper_parameters:
+      sparse_inputs_slots: 27
+      sparse_feature_number: 1000001
+      sparse_feature_dim: 8
+      dense_input_dim: 13
+      fc_sizes: [512, 256, 128, 32] 
+      learning_rate: 0.001
+
+  save:
+    increment:
+      dirname: "models_for_increment"
+      epoch_interval: 2
+      save_last: True
+    inference:
+      dirname: "models_for_inference"
+      epoch_interval: 4
+      save_last: True
+
+evaluate:
+  batch_size: 32
+  train_thread_num: 12
+  reader: "reader.py"
+
+
-- 
GitLab