.run_ce.sh 787 字节
Newer Older
B
kpi fix  
baiyfbupt 已提交
1 2 3
###!/bin/bash
####This file is only used for continuous evaluation.

B
baiyf 已提交
4 5 6 7 8
export MKL_NUM_THREADS=1
export OMP_NUM_THREADS=1

if [ ! -d "/root/.cache/paddle/dataset/pascalvoc" ];then
    mkdir -p /root/.cache/paddle/dataset/pascalvoc
B
baiyfbupt 已提交
9
    ./data/pascalvoc/download.sh
B
add ce  
baiyfbupt 已提交
10
    cp -r ./data/pascalvoc/. /home/.cache/paddle/dataset/pascalvoc
B
baiyf 已提交
11
fi
B
kpi fix  
baiyfbupt 已提交
12 13 14

cudaid=${object_detection_cudaid:=0}
export CUDA_VISIBLE_DEVICES=$cudaid
Q
qingqing01 已提交
15
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
B
kpi fix  
baiyfbupt 已提交
16

B
Bai Yifan 已提交
17
cudaid=${object_detection_cudaid_m:=0,1,2,3}
B
kpi fix  
baiyfbupt 已提交
18
export CUDA_VISIBLE_DEVICES=$cudaid
Q
qingqing01 已提交
19
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