Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

train_network.py: error: argument --caption_dropout_every_n_epochs: invalid int value: '1.0' #335

Closed
ruSauron opened this issue Mar 8, 2023 · 2 comments

Comments

@ruSauron
Copy link

ruSauron commented Mar 8, 2023

Steps to reproduce:

  1. Launch UI
  2. Set Training parameters -> Advanced Configuration -> Dropout caption every n epochs to anything larger than 0, for example 1.
  3. Start training model

Command line option --caption_dropout_every_n_epochs will be set to "1.0" instead of 1 which cause an error.

Current behaviour:
--caption_dropout_every_n_epochs="1.0"

Correct behaviour:
--caption_dropout_every_n_epochs=1

Error log:
train_network.py: error: argument --caption_dropout_every_n_epochs: invalid int value: '1.0' Traceback (most recent call last): File "D:\Programs\Python310\lib\runpy.py", line 196, in _run_module_as_main return _run_code(code, main_globals, None, File "D:\Programs\Python310\lib\runpy.py", line 86, in _run_code exec(code, run_globals) File "_PathTo_\kohya_ss\venv\Scripts\accelerate.exe\__main__.py", line 7, in <module> File "_PathTo_\kohya_ss\venv\lib\site-packages\accelerate\commands\accelerate_cli.py", line 45, in main args.func(args) File "_PathTo_\kohya_ss\venv\lib\site-packages\accelerate\commands\launch.py", line 1104, in launch_command simple_launcher(args) File "_PathTo_\kohya_ss\venv\lib\site-packages\accelerate\commands\launch.py", line 567, in simple_launcher raise subprocess.CalledProcessError(returncode=process.returncode, cmd=cmd) subprocess.CalledProcessError: Command '['_PathTo_\\kohya_ss\\venv\\Scripts\\python.exe', 'train_network.py', '_YourTrainingOptions_]' returned non-zero exit status 2.

@bmaltais
Copy link
Owner

bmaltais commented Mar 8, 2023

I will convert the field to only accept integer and use that instead of a float... That you for reporting.

@bmaltais
Copy link
Owner

bmaltais commented Mar 9, 2023

The issue is fixed in the latest dev branch commit

@bmaltais bmaltais mentioned this issue Mar 9, 2023
@bmaltais bmaltais reopened this Mar 10, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants