Skip to content
Snippets Groups Projects

Add indication of lowest validation loss epochs on loss plots

Merged André Anjos requested to merge issue-70 into main
@@ -255,7 +255,7 @@ class LoggingCallback(lightning.pytorch.Callback):
f"interval to a suitable value, so it allows some measures "
f"to be performed. Note this is only possible if the time "
f"to log a single measurement point is smaller than the "
f"time it takes to train a single epoch."
f"time it takes to **train** a single epoch."
)
else:
for metric_name, metric_value in aggregate(metrics).items():
@@ -308,7 +308,7 @@ class LoggingCallback(lightning.pytorch.Callback):
f"interval to a suitable value, so it allows some measures "
f"to be performed. Note this is only possible if the time "
f"to log a single measurement point is smaller than the "
f"time it takes to train a single epoch."
f"time it takes to **validate** a single epoch."
)
else:
for metric_name, metric_value in aggregate(metrics).items():
Loading