diff --git a/fluid/image_classification/.run_ce.sh b/fluid/image_classification/.run_ce.sh index eee6cc08f5eadbfbe48827712de33497ea89d429..516807b9e18e9e72c2a7169ccbf5390816d11430 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 363bcae789c29fd212dcffe22e72cf7aac7b665a..4f30bfefeee5250b64f0a51ef8509a488f7689b8 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,