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
3 files
+ 142
56
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -40,9 +40,9 @@ class LoggingCallback(lightning.pytorch.Callback):
super().__init__()
# timers
self._start_training_time = 0.0
self._start_training_epoch_time = 0.0
self._start_validation_epoch_time = 0.0
self._start_training_time = time.time()
self._start_training_epoch_time = time.time()
self._start_validation_epoch_time = time.time()
# log accumulators for a single flush at each training cycle
self._to_log: dict[str, float] = {}
Loading