Update epoch counting

zincware · Jul 24, 2023 · 5d403a4 · 5d403a4
1 parent 62313d8
commit 5d403a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = []