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

Update CE.

上级 0d2a35f1
...@@ -4,8 +4,8 @@ ...@@ -4,8 +4,8 @@
export FLAGS_cudnn_deterministic=True export FLAGS_cudnn_deterministic=True
cudaid=${object_detection_cudaid:=0} cudaid=${object_detection_cudaid:=0}
export CUDA_VISIBLE_DEVICES=$cudaid 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} cudaid=${object_detection_cudaid_m:=0, 1, 2, 3}
export CUDA_VISIBLE_DEVICES=$cudaid 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') ...@@ -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_top1_kpi = AccKpi('test_acc_top1', 0.05, 0, desc='TOP1 ACC')
test_acc_top5_kpi = AccKpi( test_acc_top5_kpi = AccKpi(
'test_acc_top5', 0.05, 0, actived=False, desc='TOP5 ACC') '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_kpi = AccKpi(
'train_speed', 'train_speed',
0.5, 0.5,
...@@ -29,13 +29,13 @@ train_acc_top1_card4_kpi = AccKpi( ...@@ -29,13 +29,13 @@ train_acc_top1_card4_kpi = AccKpi(
train_acc_top5_card4_kpi = AccKpi( train_acc_top5_card4_kpi = AccKpi(
'train_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC') 'train_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC')
train_cost_card4_kpi = CostKpi( 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_kpi = AccKpi(
'test_acc_top1_card4', 0.05, 0, desc='TOP1 ACC') 'test_acc_top1_card4', 0.05, 0, desc='TOP1 ACC')
test_acc_top5_card4_kpi = AccKpi( test_acc_top5_card4_kpi = AccKpi(
'test_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC') 'test_acc_top5_card4', 0.05, 0, actived=False, desc='TOP5 ACC')
test_cost_card4_kpi = CostKpi( 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_kpi = AccKpi(
'train_speed_card4', 'train_speed_card4',
0.5, 0.5,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册