add 4card kpis

fluid/object_detection/_ce.py

@@ -10,12 +10,16 @@ from kpi import CostKpi, DurationKpi, AccKpi
 train_cost_kpi = CostKpi('train_cost', 0.02, actived=True)
 test_acc_kpi = AccKpi('test_acc', 0.01, actived=True)
 train_speed_kpi = AccKpi('train_speed', 0.2, actived=True)
+train_cost_card4_kpi = CostKpi('train_cost_card4', 0.02, actived=True)
+test_acc_card4_kpi = AccKpi('test_acc_card4', 0.01, actived=True)
 train_speed_card4_kpi = AccKpi('train_speed_card4', 0.2, actived=True)
 
 tracking_kpis = [
     train_cost_kpi,
     test_acc_kpi,
     train_speed_kpi,
+    train_cost_card4_kpi,
+    test_acc_card4_kpi,
     train_speed_card4_kpi,
 ]
 
@@ -63,6 +67,6 @@ def log_to_ce(log):
 if __name__ == '__main__':
     log = sys.stdin.read()
     print("*****")
-    print log
+    print(log)
     print("****")
     log_to_ce(log)

fluid/object_detection/train.py

@@ -197,9 +197,9 @@ def train(args,
                 print ("kpis    test_acc          %s" % mean_map)
                 print ("kpis    train_speed       %s" % (total_time / epoch_idx))
             else:
-                print ("kpis    train_cost_card%s   %s" % (gpu_num, train_avg_loss))
-                print ("kpis    test_acc_card%s     %s" % (gpu_num, mean_map))
-                print ("kpis    train_speed_card%s  %f" % (gpu_num, total_time / epoch_idx))
+                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 / epoch_idx))
 
 
         if pass_id % 10 == 0 or pass_id == num_passes - 1: