diff --git a/fluid/image_classification/_ce.py b/fluid/image_classification/_ce.py
index 7ce288031a6d3015427361e47989b3f04394f3ca..bda5d88e9f84176490596d7b6094979dad705398 100644
--- a/fluid/image_classification/_ce.py
+++ b/fluid/image_classification/_ce.py
@@ -19,7 +19,7 @@ test_acc_top1_kpi = AccKpi(
 test_acc_top5_kpi = AccKpi(
     'test_acc_top5', 0.02, 0, actived=True, desc='TOP5 ACC')
 test_cost_kpi = CostKpi('test_cost', 0.02, 0, actived=True, desc='train cost')
-train_speed_kpi = AccKpi(
+train_speed_kpi = DurationKpi(
     'train_speed',
     0.05,
     0,
@@ -38,7 +38,7 @@ test_acc_top5_card4_kpi = AccKpi(
     'test_acc_top5_card4', 0.02, 0, actived=True, desc='TOP5 ACC')
 test_cost_card4_kpi = CostKpi(
     'test_cost_card4', 0.02, 0, actived=True, desc='train cost')
-train_speed_card4_kpi = AccKpi(
+train_speed_card4_kpi = DurationKpi(
     'train_speed_card4',
     0.05,
     0,