From a0ac133987a925df1907f3804ccb3cbc32b763b7 Mon Sep 17 00:00:00 2001
From: typhoonzero <typhoonzero1986@gmail.com>
Date: Tue, 16 Jan 2018 14:58:22 +0800
Subject: [PATCH] update job

---
 benchmark/cluster/v2/trainer.yaml | 6 ++++--
 benchmark/cluster/v2/vgg16.py     | 4 ++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/benchmark/cluster/v2/trainer.yaml b/benchmark/cluster/v2/trainer.yaml
index 33c95df365a..a4958b22788 100644
--- a/benchmark/cluster/v2/trainer.yaml
+++ b/benchmark/cluster/v2/trainer.yaml
@@ -21,6 +21,8 @@ spec:
         env:
         - name: PADDLE_JOB_NAME
           value: vgg16job
+        - name: OMP_NUM_THREADS
+          value: "1"
         - name: TRAINERS
           value: "20"
         - name: PSERVERS
@@ -36,7 +38,7 @@ spec:
         - name: PADDLE_INIT_NICS
           value: "xgbe0"
         - name: PADDLE_INIT_TRAINER_COUNT
-          value: "1"
+          value: "2"
         - name: PADDLE_INIT_PORTS_NUM
           value: "1"
         - name: PADDLE_INIT_PORTS_NUM_FOR_SPARSE
@@ -44,7 +46,7 @@ spec:
         - name: PADDLE_INIT_NUM_GRADIENT_SERVERS
           value: "20"
         - name: PADDLE_INIT_NUM_PASSES
-          value: "1"
+          value: "2"
         - name: PADDLE_INIT_USE_GPU
           value: "0"
         - name: LD_LIBRARY_PATH
diff --git a/benchmark/cluster/v2/vgg16.py b/benchmark/cluster/v2/vgg16.py
index 8644a547b33..85502c38e4d 100644
--- a/benchmark/cluster/v2/vgg16.py
+++ b/benchmark/cluster/v2/vgg16.py
@@ -74,14 +74,14 @@ def vgg19(input, class_dim):
 
 
 def main():
-    paddle.init(use_gpu=False, trainer_count=1)
+    paddle.init(use_gpu=False)
     image = paddle.layer.data(
         name="image", type=paddle.data_type.dense_vector(DATA_DIM))
     lbl = paddle.layer.data(
         name="label", type=paddle.data_type.integer_value(CLASS_DIM))
 
     extra_layers = None
-    learning_rate = 0.01
+    learning_rate = 1e-3
     out = vgg16(image, class_dim=CLASS_DIM)
     cost = paddle.layer.classification_cost(input=out, label=lbl)
 
-- 
GitLab