diff --git a/PaddleCV/rcnn/.run_ce.sh b/PaddleCV/rcnn/.run_ce.sh index e0e8fd52bef805177a27fc0e5c009e395016a120..8062ee1cef875897fccb4220a6285a522acef961 100755 --- a/PaddleCV/rcnn/.run_ce.sh +++ b/PaddleCV/rcnn/.run_ce.sh @@ -13,5 +13,5 @@ FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=datase 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=500 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn --learning_rate-0.005 | python _ce.py +FLAGS_benchmark=true python train.py --model_save_dir=output/ --data_dir=dataset/coco/ --max_iter=500 --enable_ce --pretrained_model=./imagenet_resnet50_fusebn --learning_rate=0.005 | python _ce.py