Skip to content
Snippets Groups Projects
Commit 725ba51b authored by Niclas Steve Eich's avatar Niclas Steve Eich
Browse files

Adjusted Early-Stopping callback verbosity

parent 91020703
No related branches found
No related tags found
No related merge requests found
......@@ -937,7 +937,7 @@ class EarlyStopping(PatientTracker):
if self.restore_best_weights and self.best_weights is not None:
self.model.set_weights(self.best_weights)
if self.verbose > 0:
print("Restoring model weights from the end of the best epoch.")
print("EarlyStopping-Callback: Restoring model weights from the end of the best epoch.")
def on_train_end(self, logs=None):
hist = self.model.history
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment