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