From 840e9beb26687887336c1cb8a26d8abd54b059dd Mon Sep 17 00:00:00 2001 From: liu zhengxi <380185688@qq.com> Date: Tue, 8 Dec 2020 21:30:41 +0800 Subject: [PATCH] update readme of transformer benchmark (#4997) --- PaddleNLP/benchmark/transformer/README.md | 2 +- PaddleNLP/examples/machine_translation/transformer/README.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/PaddleNLP/benchmark/transformer/README.md b/PaddleNLP/benchmark/transformer/README.md index 5f2d2801..77fb2011 100644 --- a/PaddleNLP/benchmark/transformer/README.md +++ b/PaddleNLP/benchmark/transformer/README.md @@ -20,5 +20,5 @@ python3 train.py ``` shell cd dygraph/ export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 -python3 -m paddle.distributed.launch --selected_gpus=0,1,2,3,4,5,6,7 train.py +python3 -m paddle.distributed.launch --gpus "0,1,2,3,4,5,6,7" train.py ``` diff --git a/PaddleNLP/examples/machine_translation/transformer/README.md b/PaddleNLP/examples/machine_translation/transformer/README.md index edf29a72..a4181bd7 100644 --- a/PaddleNLP/examples/machine_translation/transformer/README.md +++ b/PaddleNLP/examples/machine_translation/transformer/README.md @@ -66,7 +66,7 @@ python train.py ```sh export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 -python -m paddle.distributed.launch --selected_gpus=0,1,2,3,4,5,6,7 train.py +python -m paddle.distributed.launch --gpus "0,1,2,3,4,5,6,7" train.py ``` -- GitLab