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