From 8c1d6e8542add08e08e553854b0e07a022b4b9db Mon Sep 17 00:00:00 2001 From: chenxuyi Date: Sun, 29 Sep 2019 11:04:32 +0800 Subject: [PATCH] fix script --- script/zh_task/ernie_base/run_cmrc2018.sh | 2 +- script/zh_task/ernie_base/run_drcd.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/script/zh_task/ernie_base/run_cmrc2018.sh b/script/zh_task/ernie_base/run_cmrc2018.sh index b9614a3..d4967f9 100644 --- a/script/zh_task/ernie_base/run_cmrc2018.sh +++ b/script/zh_task/ernie_base/run_cmrc2018.sh @@ -2,7 +2,7 @@ set -eux export FLAGS_eager_delete_tensor_gb=0 export FLAGS_sync_nccl_allreduce=1 -export CUDA_VISIBLE_DEVICES=0,1,2,3 +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./finetune_launch.py \ --nproc_per_node 8 \ diff --git a/script/zh_task/ernie_base/run_drcd.sh b/script/zh_task/ernie_base/run_drcd.sh index 372f8b2..07bda03 100644 --- a/script/zh_task/ernie_base/run_drcd.sh +++ b/script/zh_task/ernie_base/run_drcd.sh @@ -2,7 +2,7 @@ set -eux export FLAGS_eager_delete_tensor_gb=0 export FLAGS_sync_nccl_allreduce=1 -export CUDA_VISIBLE_DEVICES=0,1,2,3 +export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python ./finetune_launch.py \ --nproc_per_node 8 \ -- GitLab