yammed
This commit is contained in:
parent
96b7f9d2cc
commit
f34cc382c5
|
@ -1641,9 +1641,9 @@ def optimize_training_settings( **kwargs ):
|
||||||
if not os.path.exists(get_halfp_model_path()):
|
if not os.path.exists(get_halfp_model_path()):
|
||||||
convert_to_halfp()
|
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
|
return settings, messages
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user