Fix LogEventStore end epoch log (#1314)

tracel-ai · Feb 19, 2024 · 80f3cdd · 80f3cdd
1 parent 9e61b66
commit 80f3cdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/burn-train/src/metric/store/log.rs b/burn-train/src/metric/store/log.rs
@@ -44,7 +44,7 @@ impl EventStore for LogEventStore {
                 Split::Valid => self
                     .loggers_valid
                     .iter_mut()
-                    .for_each(|logger| logger.end_epoch(epoch + 1)),
+                    .for_each(|logger| logger.end_epoch(epoch)),
             },
         }
     }