Skip to content

Commit

Permalink
Use zero as minimum LR for polynomial scheduler (#2410)
Browse files Browse the repository at this point in the history
  • Loading branch information
b-fission authored Apr 29, 2024
1 parent 72927fb commit fbd0d1a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion kohya_gui/class_basic_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -323,7 +323,7 @@ def init_scheduler_controls(self) -> None:
# Initialize the learning rate scheduler power textbox
self.lr_scheduler_power = gr.Number(
label="LR power",
minimum=1.0,
minimum=0.0,
step=0.01,
info="Polynomial power for polynomial scheduler",
value=self.config.get("basic.lr_scheduler_power", 1.0),
Expand Down

0 comments on commit fbd0d1a

Please sign in to comment.