diff --git a/script/zh_task/ernie_base/run_cmrc2018.sh b/script/zh_task/ernie_base/run_cmrc2018.sh index b9614a328e5fcfe23bf6139f37e57d468e528648..d4967f9a630e3f33b5dd8eb43c6fc925207b4a7d 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 372f8b20b4b79ead19ca69003942acc8bee8a475..07bda03d63c8e232240a5ac717dbb230bc423970 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 \