add a non-existing value to avg values

utils/generic_utils.py

@@ -197,6 +197,11 @@ class KeepAverage():
         self.iters[name] = init_iter
 
     def update_value(self, name, value, weighted_avg=False):
+        if name not in self.avg_values:
+            # add value if not exist before
+            self.add_value(name, init_val=value)
+        else:
+            # else update existing value
             if weighted_avg:
                 self.avg_values[name] = 0.99 * self.avg_values[name] + 0.01 * value
                 self.iters[name] += 1