From 541b42e6fb7a5f4adaaad96251659e3bc9591b9d Mon Sep 17 00:00:00 2001 From: typhoonzero Date: Fri, 19 Jan 2018 13:56:42 +0800 Subject: [PATCH] fix style --- benchmark/cluster/v2/trainer.yaml | 2 +- benchmark/cluster/v2/vgg16.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/benchmark/cluster/v2/trainer.yaml b/benchmark/cluster/v2/trainer.yaml index 33c95df365a..75fffc64b00 100644 --- a/benchmark/cluster/v2/trainer.yaml +++ b/benchmark/cluster/v2/trainer.yaml @@ -28,7 +28,7 @@ spec: - name: TOPOLOGY value: "" - name: ENTRY - value: "cd /workspace && python /workspace/vgg16.py" + value: "cd /workspace && MKL_NUM_THREADS=1 python /workspace/vgg16.py" - name: TRAINER_PACKAGE value: "/workspace" - name: PADDLE_INIT_PORT diff --git a/benchmark/cluster/v2/vgg16.py b/benchmark/cluster/v2/vgg16.py index 91894932761..59e3997d784 100644 --- a/benchmark/cluster/v2/vgg16.py +++ b/benchmark/cluster/v2/vgg16.py @@ -144,4 +144,3 @@ def main(): if __name__ == '__main__': main() - -- GitLab