提交 0441d0b6 编写于 作者: D Dang Qingqing

Update CE.

上级 0d2a35f1
......@@ -4,8 +4,8 @@
export FLAGS_cudnn_deterministic=True
cudaid=${object_detection_cudaid:=0}
export CUDA_VISIBLE_DEVICES=$cudaid
python train.py --batch_size=64 --num_epochs=5 --enable_ce=True | python _ce.py
python train.py --batch_size=64 --num_epochs=10 --enable_ce=True | python _ce.py
cudaid=${object_detection_cudaid_m:=0, 1, 2, 3}
export CUDA_VISIBLE_DEVICES=$cudaid
python train.py --batch_size=128 --num_epochs=5 --enable_ce=True | python _ce.py
python train.py --batch_size=64 --num_epochs=10 --enable_ce=True | python _ce.py
......@@ -16,7 +16,7 @@ train_cost_kpi = CostKpi('train_cost', 0.5, 0, actived=True, desc='train cost')
test_acc_top1_kpi = AccKpi('test_acc_top1', 0.05, 0, desc='TOP1 ACC')
test_acc_top5_kpi = AccKpi(
'test_acc_top5', 0.05, 0, actived=False, desc='TOP5 ACC')
test_cost_kpi = CostKpi('test_cost', 0.05, 0, actived=True, desc='train cost')
test_cost_kpi = CostKpi('test_cost', 0.5, 0, actived=True, desc='train cost')
train_speed_kpi = AccKpi(
'train_speed',
0.5,
......@@ -29,13 +29,13 @@ train_acc_top1_card4_kpi = AccKpi(
train_acc_top5_card4_kpi = AccKpi(
'train_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC')
train_cost_card4_kpi = CostKpi(
'train_cost_kpi', 0.05, 0, actived=True, desc='train cost')
'train_cost_card4', 0.5, 0, actived=True, desc='train cost')
test_acc_top1_card4_kpi = AccKpi(
'test_acc_top1_card4', 0.05, 0, desc='TOP1 ACC')
test_acc_top5_card4_kpi = AccKpi(
'test_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC')
test_cost_card4_kpi = CostKpi(
'test_cost_card4', 0.05, 0, actived=True, desc='train cost')
'test_cost_card4', 0.5, 0, actived=True, desc='train cost')
train_speed_card4_kpi = AccKpi(
'train_speed_card4',
0.5,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册