Skip to content

Update train.py #654

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update train.py #654

wants to merge 1 commit into from

Conversation

pctablet505
Copy link

While using the train APIs in my own code, it is giving error that both parameters shall be float
This change fixes the issue.

StyleGAN3: File "/usr/local/lib/python3.11/dist-packages/click/core.py", line 1161, in __call__
2025-04-30 10:01:21,396 - INFO - [models_init] - StyleGAN3: return self.main(*args, **kwargs)
2025-04-30 10:01:21,396 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,396 - INFO - [models_init] - StyleGAN3: File "/usr/local/lib/python3.11/dist-packages/click/core.py", line 1082, in main
2025-04-30 10:01:21,396 - INFO - [models_init] - StyleGAN3: rv = self.invoke(ctx)
2025-04-30 10:01:21,397 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,397 - INFO - [models_init] - StyleGAN3: File "/usr/local/lib/python3.11/dist-packages/click/core.py", line 1443, in invoke
2025-04-30 10:01:21,397 - INFO - [models_init] - StyleGAN3: return ctx.invoke(self.callback, **ctx.params)
2025-04-30 10:01:21,397 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: File "/usr/local/lib/python3.11/dist-packages/click/core.py", line 788, in invoke
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: return __callback(*args, **kwargs)
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/train.py", line 281, in main
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: launch_training(c=c, desc=desc, outdir=opts.outdir, dry_run=opts.dry_run)
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/train.py", line 96, in launch_training
2025-04-30 10:01:21,398 - INFO - [models_init] - StyleGAN3: subprocess_fn(rank=0, c=c, temp_dir=temp_dir)
2025-04-30 10:01:21,399 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/train.py", line 47, in subprocess_fn
2025-04-30 10:01:21,399 - INFO - [models_init] - StyleGAN3: training_loop.training_loop(rank=rank, **c)
2025-04-30 10:01:21,399 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/training/training_loop.py", line 197, in training_loop
2025-04-30 10:01:21,399 - INFO - [models_init] - StyleGAN3: opt = dnnlib.util.construct_class_by_name(params=module.parameters(), **opt_kwargs) # subclass of torch.optim.Optimizer
2025-04-30 10:01:21,400 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,400 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/dnnlib/util.py", line 303, in construct_class_by_name
2025-04-30 10:01:21,400 - INFO - [models_init] - StyleGAN3: return call_func_by_name(*args, func_name=class_name, **kwargs)
2025-04-30 10:01:21,400 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,400 - INFO - [models_init] - StyleGAN3: File "/content/cv_face/models/stylegan3/dnnlib/util.py", line 298, in call_func_by_name
2025-04-30 10:01:21,401 - INFO - [models_init] - StyleGAN3: return func_obj(*args, **kwargs)
2025-04-30 10:01:21,401 - INFO - [models_init] - StyleGAN3: ^^^^^^^^^^^^^^^^^^^^^^^^^
2025-04-30 10:01:21,401 - INFO - [models_init] - StyleGAN3: File "/usr/local/lib/python3.11/dist-packages/torch/optim/adam.py", line 71, in __init__
2025-04-30 10:01:21,402 - INFO - [models_init] - StyleGAN3: raise ValueError("betas must be either both floats or both Tensors")
2025-04-30 10:01:21,402 - INFO - [models_init] - StyleGAN3: ValueError: betas must be either both floats or both Tensors
2025-04-30 10:01:22,292 - ERROR - [models_init] - StyleGAN3 training failed with return code 1
2025-04-30 10:01:22,292 - INFO - [models_init] - StyleGAN3 finetuning task finished.

@pctablet505 pctablet505 mentioned this pull request May 1, 2025
Open
@pctablet505
Copy link
Author

pctablet505 commented May 1, 2025 •
edited
Loading

Fixes #655.

@nurpax, @tkarras, @jannehellsten may you please review it.

@Mininaut
Copy link

Mininaut commented May 26, 2025 •
edited
Loading

tf i click

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
1 more reviewer

@Mininaut Mininaut Mininaut approved these changes

Reviewers whose approvals may not affect merge requirements
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@pctablet505 @Mininaut