.run_ce.sh 718 字节
Newer Older
Z
zhengya01 已提交
1 2 3 4 5 6
#!/bin/bash

export MKL_NUM_THREADS=1
export OMP_NUM_THREADS=1


Z
add ce  
zhengya01 已提交
7
cudaid=${face_detection:=0} # use 0-th card as default
Z
zhengya01 已提交
8 9 10 11 12
export CUDA_VISIBLE_DEVICES=$cudaid

FLAGS_benchmark=true  python train.py --dataset=coco --num_epochs=2 --batch_num=10 --enable_ce | python _ce.py


Z
add ce  
zhengya01 已提交
13
cudaid=${face_detection_4:=0,1,2,3} # use 0,1,2,3 card as default
Z
zhengya01 已提交
14 15 16
export CUDA_VISIBLE_DEVICES=$cudaid

FLAGS_benchmark=true  python train.py --dataset=coco --num_epochs=2 --batch_num=10 --enable_ce | python _ce.py
Z
add ce  
zhengya01 已提交
17 18


Z
add ce  
zhengya01 已提交
19
cudaid=${face_detection_8:=0,1,2,3,4,5,6,7} # use 0,1,2,3,4,5,6,7 card as default
Z
add ce  
zhengya01 已提交
20 21 22 23 24
export CUDA_VISIBLE_DEVICES=$cudaid

FLAGS_benchmark=true  python train.py --dataset=coco --num_epochs=2 --batch_num=10 --enable_ce | python _ce.py