diff --git a/fluid/PaddleCV/human_pose_estimation/.run_ce.sh b/fluid/PaddleCV/human_pose_estimation/.run_ce.sh index 434d0edc5f6cffcd32ffbdd8b3e0a71637ac5380..593cbe1f3aa15e4ff20810bfd17b5c1ea9e9dca5 100755 --- a/fluid/PaddleCV/human_pose_estimation/.run_ce.sh +++ b/fluid/PaddleCV/human_pose_estimation/.run_ce.sh @@ -16,7 +16,7 @@ export CUDA_VISIBLE_DEVICES=$cudaid FLAGS_benchmark=true python train.py --dataset=coco --num_epochs=2 --batch_num=10 --enable_ce | python _ce.py -cudaid=${face_detection_8:=0,1,2,3,4,5,6,7} # use 0,1,2,3 card as default +cudaid=${face_detection_8:=0,1,2,3,4,5,6,7} # use 0,1,2,3,4,5,6,7 card as default export CUDA_VISIBLE_DEVICES=$cudaid FLAGS_benchmark=true python train.py --dataset=coco --num_epochs=2 --batch_num=10 --enable_ce | python _ce.py