1580158
/
kohya_ss
mirror of https://github.com/bmaltais/kohya_ss
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge e1d53ea5cc into 4161d1d80a

pull/3456/merge
Ian 2025-11-03 13:28:32 +01:00 committed by GitHub
parent 4161d1d80a e1d53ea5cc
commit f4bc59b4db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
kohya_gui/finetune_gui.py
View File

@ -875,12 +875,14 @@ def train_model(
log.info(max_train_steps_info)
# Calculate lr_warmup_steps
lr_warmup = float(lr_warmup) if lr_warmup not in ("", None) else 0
lr_warmup_steps = int(lr_warmup_steps) if lr_warmup_steps not in ("", None) else 0
if lr_warmup_steps > 0:
lr_warmup_steps = int(lr_warmup_steps)
if lr_warmup > 0:
log.warning("Both lr_warmup and lr_warmup_steps are set. lr_warmup_steps will be used.")
elif lr_warmup != 0:
lr_warmup_steps = lr_warmup / 100
lr_warmup_steps = int(lr_warmup * max_train_steps / 100) if max_train_steps > 0 else 0
else:
lr_warmup_steps = 0