.run_ce.sh 789 字节
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 15 16 17 18 19

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

cudaid=${object_detection_cudaid:=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