提交 bef4c677 编写于 作者: Q qingqing01 提交者: Bai Yifan

Fix CE (#1238)

上级 872a6eff
......@@ -12,8 +12,8 @@ fi
cudaid=${object_detection_cudaid:=0}
export CUDA_VISIBLE_DEVICES=$cudaid
FLAGS_benchmark=true python train.py --enable_ce=True --batch_size=64 --num_passes=2 --data_dir=/root/.cache/paddle/dataset/pascalvoc/ | python _ce.py
FLAGS_benchmark=true python train.py --enable_ce=True --batch_size=64 --epoc_num=2 --data_dir=/root/.cache/paddle/dataset/pascalvoc/ | python _ce.py
cudaid=${object_detection_cudaid_m:=0,1,2,3}
export CUDA_VISIBLE_DEVICES=$cudaid
FLAGS_benchmark=true python train.py --enable_ce=True --batch_size=64 --num_passes=2 --data_dir=/root/.cache/paddle/dataset/pascalvoc/ | python _ce.py
FLAGS_benchmark=true python train.py --enable_ce=True --batch_size=64 --epoc_num=2 --data_dir=/root/.cache/paddle/dataset/pascalvoc/ | python _ce.py
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册