Super-small fix stops us confusing Keras console logging by modifying its logs (huggingface#15373)

Rocketknight1 · web-flow · commit 2de90beeeb58 · 2022-01-27T15:43:43.000Z
diff --git a/src/transformers/keras_callbacks.py b/src/transformers/keras_callbacks.py
@@ -324,6 +324,7 @@ def on_train_batch_end(self, batch, logs=None):
             )
 
     def on_epoch_end(self, epoch, logs=None):
+        logs = logs.copy()  # Don't accidentally write things that Keras will read later
         if "epoch" not in logs:
             logs["epoch"] = epoch
         self.training_history.append(logs)

Original file line number	Diff line number	Diff line change
`@@ -324,6 +324,7 @@ def on_train_batch_end(self, batch, logs=None):`
`324`	`324`	`)`
`325`	`325`
`326`	`326`	`def on_epoch_end(self, epoch, logs=None):`
	`327`	`+ logs = logs.copy() # Don't accidentally write things that Keras will read later`
`327`	`328`	`if "epoch" not in logs:`
`328`	`329`	`logs["epoch"] = epoch`
`329`	`330`	`self.training_history.append(logs)`