背景
用户目前只能通过「其他设置 → ui_custom_params」写入危险覆盖参数,才能使用带最小学习率的余弦调度,例如:
lr_scheduler = "cosine_with_min_lr"
lr_scheduler_min_lr_ratio = 0.1
lr_scheduler_num_cycles = 1
lr_warmup_steps = 0
其中 lr_warmup_steps 已在正式表单中;前三项应一并作为正式 UI 能力暴露,避免依赖自定义覆盖。
现状
| 参数 |
sd-scripts / Kohya |
当前 UI |
lr_scheduler=cosine_with_min_lr |
✅ train_util.get_scheduler_fix 已支持 |
❌ 下拉未包含 |
lr_scheduler_min_lr_ratio |
✅ 官方 CLI |
❌ schema 未暴露 |
lr_scheduler_num_cycles |
✅(cosine_with_min_lr 内部按 num_cycles/2) |
⚠️ 仅 cosine_with_restarts 时显示 |
lr_warmup_steps |
✅ |
✅ 已有 |
结论:引擎已通,缺产品层暴露(共享 LR_OPTIMIZER / 各页 schema)。
目标(合并为一个 Issue)
- 学习率调度器选项增加
cosine_with_min_lr
- 选中后显示
lr_scheduler_min_lr_ratio(建议默认 0.1,可配置)
- 在适用调度器下显示
lr_scheduler_num_cycles(含 cosine_with_restarts 与 cosine_with_min_lr),并在描述中注明 cosine_with_min_lr 与 restarts 的语义差异(训练脚本内 /2)
lr_warmup_steps 保持现有字段;文档/描述可点出与 cosine_with_min_lr 的常用组合
覆盖使用共享 LR schema 的主路径(SDXL / Flux / Anima 等 Kohya 系);Krea2(Musubi 白名单已含相关字段)、Anima Fast 需在实现时做一次可用性确认,不支持则隐藏或标注。
非目标
- 引入全新调度算法实现(上游已有)
- 通过
ui_custom_params 以外的「任意 scheduler 字符串」万能入口
- 改变默认调度器(仍可保持现有默认,如
cosine_with_restarts / constant)
验收
优先级
P2(体验补齐;已有 custom_params workaround)
背景
用户目前只能通过「其他设置 →
ui_custom_params」写入危险覆盖参数,才能使用带最小学习率的余弦调度,例如:其中
lr_warmup_steps已在正式表单中;前三项应一并作为正式 UI 能力暴露,避免依赖自定义覆盖。现状
lr_scheduler=cosine_with_min_lrtrain_util.get_scheduler_fix已支持lr_scheduler_min_lr_ratiolr_scheduler_num_cyclescosine_with_min_lr内部按num_cycles/2)cosine_with_restarts时显示lr_warmup_steps结论:引擎已通,缺产品层暴露(共享
LR_OPTIMIZER/ 各页 schema)。目标(合并为一个 Issue)
cosine_with_min_lrlr_scheduler_min_lr_ratio(建议默认0.1,可配置)lr_scheduler_num_cycles(含cosine_with_restarts与cosine_with_min_lr),并在描述中注明cosine_with_min_lr与 restarts 的语义差异(训练脚本内/2)lr_warmup_steps保持现有字段;文档/描述可点出与cosine_with_min_lr的常用组合覆盖使用共享 LR schema 的主路径(SDXL / Flux / Anima 等 Kohya 系);Krea2(Musubi 白名单已含相关字段)、Anima Fast 需在实现时做一次可用性确认,不支持则隐藏或标注。
非目标
ui_custom_params以外的「任意 scheduler 字符串」万能入口cosine_with_restarts/constant)验收
ui_custom_params,直接选cosine_with_min_lr并设置min_lr_ratio/num_cycles/warmupcosine_with_restarts显示 cycles 的行为(扩展后仍正确)min_lr_ratio含义优先级
P2(体验补齐;已有 custom_params workaround)