From f3675a9622e290b76a2475c9b6381b845c7fce95 Mon Sep 17 00:00:00 2001
From: Jonathunky <12529409+Jonathunky@users.noreply.github.com>
Date: Tue, 31 Oct 2023 09:55:41 +0100
Subject: [PATCH] Small fix for time display

---
 u2net_train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/u2net_train.py b/u2net_train.py
index fa6681aa..8bf97752 100644
--- a/u2net_train.py
+++ b/u2net_train.py
@@ -311,8 +311,8 @@ def train_epochs(
         if sum(training_counts.values()) == 3:
             elapsed_time = time.time() - start_time
             minutes, seconds = divmod(elapsed_time, 60)
-            print(f"Expected performance is {minutes:.1f} minutes seconds per epoch.\n")
-
+            perf = minutes + (seconds / 60)
+            print(f"Expected performance is {perf:.1f} minutes per epoch.\n")
         # Increment the corresponding training count
         training_counts[key] += 1