From f34cc382c5b785dbf4565e509cbc75462447d72c Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 17 Mar 2023 18:57:36 +0000 Subject: [PATCH] yammed --- src/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/utils.py b/src/utils.py index 8fb3e98..52addfb 100755 --- a/src/utils.py +++ b/src/utils.py @@ -1641,9 +1641,9 @@ def optimize_training_settings( **kwargs ): if not os.path.exists(get_halfp_model_path()): convert_to_halfp() - settings['steps'] = int(iterations / settings['epochs']) + steps = int(iterations / settings['epochs']) - messages.append(f"For {settings['epochs']} epochs with {lines} lines in batches of {settings['batch_size']}, iterating for {iterations} steps ({settings['steps']}) steps per epoch)") + messages.append(f"For {settings['epochs']} epochs with {lines} lines in batches of {settings['batch_size']}, iterating for {iterations} steps ({steps}) steps per epoch)") return settings, messages