From 5d403a432a1256422d69527a40c8081e346969c7 Mon Sep 17 00:00:00 2001 From: SamTov Date: Mon, 24 Jul 2023 23:33:04 +0200 Subject: [PATCH] Update epoch counting --- znnl/training_strategies/simple_training.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/znnl/training_strategies/simple_training.py b/znnl/training_strategies/simple_training.py index b7e3534..4af7e8b 100644 --- a/znnl/training_strategies/simple_training.py +++ b/znnl/training_strategies/simple_training.py @@ -360,7 +360,7 @@ def train_model( state = self.model.model_state loading_bar = trange( - 1, epochs + 1, ncols=100, unit="batch", disable=self.disable_loading_bar + 0, epochs, ncols=100, unit="batch", disable=self.disable_loading_bar ) train_losses = []