From 0441d0b690484cb33542b5641437c064f9012ef4 Mon Sep 17 00:00:00 2001 From: Dang Qingqing Date: Mon, 27 Aug 2018 10:24:27 +0800 Subject: [PATCH] Update CE. --- fluid/image_classification/.run_ce.sh | 4 ++-- fluid/image_classification/_ce.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/fluid/image_classification/.run_ce.sh b/fluid/image_classification/.run_ce.sh index eee6cc08..516807b9 100755 --- a/fluid/image_classification/.run_ce.sh +++ b/fluid/image_classification/.run_ce.sh @@ -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 diff --git a/fluid/image_classification/_ce.py b/fluid/image_classification/_ce.py index 363bcae7..4f30bfef 100644 --- a/fluid/image_classification/_ce.py +++ b/fluid/image_classification/_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, -- GitLab