add ce

fluid/PaddleCV/human_pose_estimation/.run_ce.sh

@@ -4,13 +4,21 @@ export MKL_NUM_THREADS=1
 export OMP_NUM_THREADS=1
 
 
-cudaid=${face_detection:=7} # use 0-th card as default
+cudaid=${face_detection:=0} # use 0-th 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
 
 
-cudaid=${face_detection_m:=4,5,6,7} # use 0,1,2,3 card as default
+cudaid=${face_detection_4:=0,1,2,3} # use 0,1,2,3 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
+
+
+cudaid=${face_detection_8:=0,1,2,3,4,5,6,7} # use 0,1,2,3 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
+
+

fluid/PaddleCV/human_pose_estimation/_ce.py

@@ -11,12 +11,16 @@ each_pass_duration_card1_kpi = DurationKpi('each_pass_duration_card1', 0.08, 0,
 train_loss_card1_kpi = CostKpi('train_loss_card1', 0.08, 0)
 each_pass_duration_card4_kpi = DurationKpi('each_pass_duration_card4', 0.08, 0, actived=True)
 train_loss_card4_kpi = CostKpi('train_loss_card4', 0.08, 0)
+each_pass_duration_card8_kpi = DurationKpi('each_pass_duration_card8', 0.08, 0, actived=True)
+train_loss_card8_kpi = CostKpi('train_loss_card8', 0.08, 0)
 
 tracking_kpis = [
         each_pass_duration_card1_kpi,
         train_loss_card1_kpi,
         each_pass_duration_card4_kpi,
         train_loss_card4_kpi,
+        each_pass_duration_card8_kpi,
+        train_loss_card8_kpi,
         ]