From eee9ac073bc42fa43ca751f1a86a55794217194b Mon Sep 17 00:00:00 2001
From: Clemens Schmid <clemisch@tuta.io>
Date: Mon, 11 Nov 2024 17:20:49 +0100
Subject: [PATCH] Pad epoch index

---
 src/cdtools/models/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cdtools/models/base.py b/src/cdtools/models/base.py
index 180ba18..b31061d 100644
--- a/src/cdtools/models/base.py
+++ b/src/cdtools/models/base.py
@@ -706,7 +706,7 @@ class CDIModel(t.nn.Module):
             epoch = len(self.loss_history)
             dt = self.latest_iteration_time
             loss = self.loss_history[-1]
-            msg = f'Epoch {epoch} completed in {dt:0.2f} s with loss {loss:.5e}'
+            msg = f'Epoch {epoch:3d} completed in {dt:0.2f} s with loss {loss:.5e}'
         else:
             msg = 'No reconstruction iterations performed yet!'