From 9e76aa14796e04ef24c10c1080edcabf38341caf Mon Sep 17 00:00:00 2001 From: kolinwei <331911734@qq.com> Date: Wed, 13 Mar 2019 15:05:32 +0800 Subject: [PATCH] Update .run_ce.sh (#1883) * Update .run_ce.sh --- fluid/PaddleCV/rcnn/.run_ce.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/fluid/PaddleCV/rcnn/.run_ce.sh b/fluid/PaddleCV/rcnn/.run_ce.sh index af2f21da..15023b71 100755 --- a/fluid/PaddleCV/rcnn/.run_ce.sh +++ b/fluid/PaddleCV/rcnn/.run_ce.sh @@ -7,11 +7,11 @@ export OMP_NUM_THREADS=1 cudaid=${face_detection:=0} # use 0-th card as default export CUDA_VISIBLE_DEVICES=$cudaid -FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=dataset/coco/ --max_iter=10 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn | python _ce.py +FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=dataset/coco/ --max_iter=100 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn | python _ce.py cudaid=${face_detection_m:=0,1,2,3} # use 0,1,2,3 card as default export CUDA_VISIBLE_DEVICES=$cudaid -FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=dataset/coco/ --max_iter=10 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn | python _ce.py +FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=dataset/coco/ --max_iter=100 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn | python _ce.py -- GitLab