fix the issue(apache#14468):'float' object is not subscriptable

wuxun-zhang · Mar 24, 2019 · af7ab07 · af7ab07
1 parent 056fce4
commit af7ab07
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/example/ssd/train/metric.py b/example/ssd/train/metric.py
@@ -32,6 +32,21 @@ def reset(self):
         """
         override reset behavior
         """
+        if getattr(self, 'num', None) is None:
+            self.num_inst = 0
+            self.sum_metric = 0.0
+            self.global_num_inst = 0
+            self.global_sum_metric = 0.0
+        else:
+            self.num_inst = [0] * self.num
+            self.sum_metric = [0.0] * self.num
+            self.global_num_inst = [0] * self.num
+            self.global_sum_metric = [0.0] * self.num
+
+    def reset_local(self):
+        """
+        override reset_local behavior
+        """
         if getattr(self, 'num', None) is None:
             self.num_inst = 0
             self.sum_metric = 0.0