From e222caa05a81ce1d73553926cc8aa4359ffbfaaa Mon Sep 17 00:00:00 2001 From: zhengya01 Date: Wed, 9 Jan 2019 08:38:35 +0000 Subject: [PATCH] add ce --- fluid/PaddleCV/human_pose_estimation/.run_ce.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fluid/PaddleCV/human_pose_estimation/.run_ce.sh b/fluid/PaddleCV/human_pose_estimation/.run_ce.sh index 434d0edc..593cbe1f 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 -- GitLab