Update train.py

[pctablet505]

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]

Fixes #655.

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

[Mininaut]

tf i click