diff --git a/benchmark/cluster/vgg16/v2_pserver.yaml b/benchmark/cluster/vgg16/v2_pserver.yaml index dd1271e0cf399184134c06b3200ee1202c65cef0..857e2ff4557d4a638bcdeb586d9fc8978d1183df 100644 --- a/benchmark/cluster/vgg16/v2_pserver.yaml +++ b/benchmark/cluster/vgg16/v2_pserver.yaml @@ -23,13 +23,13 @@ spec: - name: PADDLE_JOB_NAME value: vgg16v2job - name: TRAINERS - value: "20" + value: "60" - name: PSERVERS value: "10" - name: TOPOLOGY value: "" - name: ENTRY - value: "python train.py" + value: "python -u train.py" - name: TRAINER_PACKAGE value: "/workspace" - name: PADDLE_INIT_PORT diff --git a/benchmark/cluster/vgg16/v2_trainer.yaml b/benchmark/cluster/vgg16/v2_trainer.yaml index 997bbc81c99ae642271b3ed4c5c141a5dfc5148a..be0f741b349c47e8f1978c0bc1d220284c96a53f 100644 --- a/benchmark/cluster/vgg16/v2_trainer.yaml +++ b/benchmark/cluster/vgg16/v2_trainer.yaml @@ -3,8 +3,8 @@ kind: Job metadata: name: vgg16v2job-trainer spec: - parallelism: 20 - completions: 20 + parallelism: 60 + completions: 60 template: metadata: labels: @@ -24,13 +24,13 @@ spec: - name: BATCH_SIZE value: "256" - name: TRAINERS - value: "20" + value: "60" - name: PSERVERS value: "10" - name: TOPOLOGY value: "" - name: ENTRY - value: "cd /workspace && MKL_NUM_THREADS=1 python /workspace/vgg16_v2.py" + value: "cd /workspace && MKL_NUM_THREADS=1 python -u /workspace/vgg16_v2.py" - name: TRAINER_PACKAGE value: "/workspace" - name: PADDLE_INIT_PORT