From d18e7d04d6f37255d61fe796fc826e6775f117c0 Mon Sep 17 00:00:00 2001 From: pkpk Date: Fri, 14 Feb 2020 10:35:22 +0800 Subject: [PATCH] Update run_classifier_multi_gpu.sh (#4279) remove the CUDA_VISIBLE_DEVICES --- dygraph/bert/run_classifier_multi_gpu.sh | 2 -- 1 file changed, 2 deletions(-) diff --git a/dygraph/bert/run_classifier_multi_gpu.sh b/dygraph/bert/run_classifier_multi_gpu.sh index 1e6ba61f..238d9e90 100755 --- a/dygraph/bert/run_classifier_multi_gpu.sh +++ b/dygraph/bert/run_classifier_multi_gpu.sh @@ -6,8 +6,6 @@ DATA_PATH="./data/glue_data/MNLI/" CKPT_PATH="./data/saved_model/mnli_models" GPU_TO_USE=0,1,2,3 -export CUDA_VISIBLE_DEVICES=$GPU_TO_USE - # start fine-tuning python -m paddle.distributed.launch --selected_gpus=$GPU_TO_USE --log_dir ./cls_log run_classifier.py \ --task_name ${TASK_NAME} \ -- GitLab