.run_ce.sh 526 字节
Newer Older
1 2 3
#!/bin/bash

# This file is only used for continuous evaluation.
D
Dang Qingqing 已提交
4
export FLAGS_cudnn_deterministic=True
B
Bai Yifan 已提交
5
BATCH_SIZE=56
6 7
cudaid=${object_detection_cudaid:=0}
export CUDA_VISIBLE_DEVICES=$cudaid
B
Bai Yifan 已提交
8
python train.py --batch_size=${BATCH_SIZE} --num_epochs=5 --enable_ce=True --lr_strategy=cosine_decay | python _ce.py
9

10
BATCH_SIZE=224
B
Bai Yifan 已提交
11
cudaid=${object_detection_cudaid_m:=0, 1, 2, 3}
12
export CUDA_VISIBLE_DEVICES=$cudaid
B
Bai Yifan 已提交
13
python train.py --batch_size=${BATCH_SIZE} --num_epochs=5 --enable_ce=True --lr_strategy=cosine_decay | python _ce.py