This commit is contained in:
mrq 2023-03-17 18:57:36 +00:00
parent 96b7f9d2cc
commit f34cc382c5

View File

@ -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