From 7cfe5ade6bc104241ec987c75ce0b5c5e87b4e98 Mon Sep 17 00:00:00 2001 From: Kurt Stolle Date: Mon, 26 Feb 2024 11:36:04 +0100 Subject: [PATCH] Merge --- sources/unipercept/engine/_engine.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sources/unipercept/engine/_engine.py b/sources/unipercept/engine/_engine.py index 35d101e..01d2f90 100644 --- a/sources/unipercept/engine/_engine.py +++ b/sources/unipercept/engine/_engine.py @@ -1333,8 +1333,10 @@ def _train_log(self, logs: dict[str, T.Any]) -> None: logs : dict[str, float] The logs to be logged. """ - logs["engine/epoch"] = round(self._state.epoch, 2) + logs["engine/epoch"] = round(self._state.epoch, 6) logs["engine/step"] = self._state.step + logs["engine/gradient_accumulation"] = self._state.gradient_accumulation + logs["engine/stage"] = self._state.stage logs["engine/epoch_step"] = self.xlr.step logs["engine/status"] = self.status @@ -1343,7 +1345,7 @@ def _train_log(self, logs: dict[str, T.Any]) -> None: self._state.log_history.append(logs) if len(self._state.log_history) > self._params.logging_history: self._state.log_history.pop(0) - self.xlr.log(logs, step=self._state.step) + self.xlr.log(logs) # , step=self._state.step) def _load_weights(self, path: Pathable, model: nn.Module) -> nn.Module: """