diff --git a/benchmark/cluster/v2/trainer.yaml b/benchmark/cluster/v2/trainer.yaml index 33c95df365..75fffc64b0 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 9189493276..59e3997d78 100644 --- a/benchmark/cluster/v2/vgg16.py +++ b/benchmark/cluster/v2/vgg16.py @@ -144,4 +144,3 @@ def main(): if __name__ == '__main__': main() -