Fill keys in CSVLogger before stop_training is checked (keras-team#8801)

Fixes keras-team#8797.
chaohuazhu · Dec 15, 2017 · c45a2c7 · c45a2c7
1 parent 0611d80
commit c45a2c7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/keras/callbacks.py b/keras/callbacks.py
@@ -965,13 +965,14 @@ def handle_value(k):
             else:
                 return k
 
+        if self.keys is None:
+            self.keys = sorted(logs.keys())
+
         if self.model.stop_training:
             # We set NA so that csv parsers do not fail for this last epoch.
             logs = dict([(k, logs[k]) if k in logs else (k, 'NA') for k in self.keys])
 
         if not self.writer:
-            self.keys = sorted(logs.keys())
-
             class CustomDialect(csv.excel):
                 delimiter = self.sep