From 654b535a0858f905b0df8874f216a60166d50691 Mon Sep 17 00:00:00 2001 From: baiyfbupt Date: Wed, 29 Aug 2018 20:28:57 +0800 Subject: [PATCH] fix ce --- fluid/object_detection/_ce.py | 8 ++++---- fluid/object_detection/train.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/fluid/object_detection/_ce.py b/fluid/object_detection/_ce.py index 43dc894b..71514891 100644 --- a/fluid/object_detection/_ce.py +++ b/fluid/object_detection/_ce.py @@ -7,12 +7,12 @@ from kpi import CostKpi, DurationKpi, AccKpi #### NOTE kpi.py should shared in models in some way!!!! -train_cost_kpi = CostKpi('train_cost', 0.2, 0, actived=True) +train_cost_kpi = CostKpi('train_cost', 0.02, 0, actived=True) test_acc_kpi = AccKpi('test_acc', 0.01, 0, actived=True) -train_speed_kpi = AccKpi('train_speed', 1, 0, actived=True) -train_cost_card4_kpi = CostKpi('train_cost_card4', 0.2, 0, actived=True) +train_speed_kpi = AccKpi('train_speed', 0.1, 0, actived=True) +train_cost_card4_kpi = CostKpi('train_cost_card4', 0.02, 0, actived=True) test_acc_card4_kpi = AccKpi('test_acc_card4', 0.01, 0, actived=True) -train_speed_card4_kpi = AccKpi('train_speed_card4', 1, 0, actived=True) +train_speed_card4_kpi = AccKpi('train_speed_card4', 0.1, 0, actived=True) tracking_kpis = [ train_cost_kpi, diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 0c2395e8..b40da930 100644 --- a/fluid/object_detection/train.py +++ b/fluid/object_detection/train.py @@ -233,14 +233,14 @@ def train(args, if devices_num == 1: print("kpis train_cost %s" % train_avg_loss) print("kpis test_acc %s" % mean_map) - print("kpis train_speed %s" % (total_time / epocs)) + print("kpis train_speed %s" % (epocs / total_time)) else: print("kpis train_cost_card%s %s" % (devices_num, train_avg_loss)) print("kpis test_acc_card%s %s" % (devices_num, mean_map)) print("kpis train_speed_card%s %f" % - (devices_num, total_time / test_epocs)) + (devices_num, test_epocs / total_time)) if pass_id % 10 == 0 or pass_id == num_passes - 1: save_model(str(pass_id), train_prog) -- GitLab