Kohya_SS 学习率自动填充指南:3 步让配置恢复正常
【免费下载链接】kohya_ss项目地址: https://gitcode.com/GitHub_Trending/ko/kohya_ss
Kohya_SS 是 Stable Diffusion 系模型微调与 LoRA 训练的图形化工具。本文解决 GUI 里学习率不自动填充、预设加载不生效的问题。读完你能独立检查 config.toml 与 presets 目录,并给不同模型填入正确学习率。
🧭 什么时候用得上
第一次打开 Kohya_SS GUI,进入 LoRA 或 Dreambooth 标签页,发现 "Text Encoder learning rate" 和 "Unet learning rate" 输入框是空的,或者填的值和你上次保存的不一样。多半是配置链路断了:config.toml 没建、格式写错,或者 presets 目录缺文件。按下面的顺序排查,十分钟能定位问题。
🛠️ 分步操作
步骤 1|确认 Kohya_SS 已安装并能启动
- 克隆仓库并进入根目录:
git clone https://gitcode.com/GitHub_Trending/ko/kohya_ss- 运行 setup.sh 或 setup.bat 完成环境安装。
- 用 gui.sh 或 gui.bat 启动 GUI,确认浏览器能打开主界面,各标签页正常渲染。
步骤 2|建立并检查 config.toml
- 在仓库根目录把 config example.toml 复制一份,改名为 config.toml。
- 打开 config example.toml 对照检查
[basic]段:应能看到 learning_rate、learning_rate_te、learning_rate_te1、learning_rate_te2 四项。 - 确认 lr_scheduler 与 optimizer 有值,示例文件里分别是 cosine 和 AdamW8bit。
- 若文件不存在,GUI 会静默回退到默认值(学习率 1e-6),输入框看起来像"没保存"。此时建好文件后重启 GUI,输入框应显示配置中的数值。
步骤 3|核对 presets 预设目录
- 打开 presets/,确认 lora 与 finetune 两个子目录存在且有 JSON 文件。
- 打开任意预设(如 SDXL - LoRA AI_characters standard v1.1.json),确认 learning_rate、text_encoder_lr、unet_lr 三个键都有值。
- 若预设文件损坏,程序会把学习率字段转成字符串再解析,解析失败时统一回落到 0.0,界面就表现为"自动填充失效"。重新获取完整仓库即可。
步骤 4|手动填入学习率并保存
- 进入对应训练标签页,在 "Text Encoder learning rate" 与 "Unet learning rate" 输入框填入下表数值。
- Unet 学习率填 0 时会回退到主学习率 learning_rate,这是设计行为,不是 bug。
- 用界面"保存"按钮把当前参数存成自定义预设到 presets 目录。下次加载时数值应保持不变。
⚠️ 新手避坑
现象:输入框一直显示 0 或空。原因:config.toml 缺失,GUI 用了内置默认值但没写回文件。对策:按步骤 2 建立 config.toml,重启 GUI 再检查。
现象:切换 SD 1.5 / SDXL 后学习率没变。原因:GUI 不会替你改学习率,它只读取配置和预设。对策:切换模型类型后,对照步骤 4 的速查表手动更新两个 LR 输入框。
现象:加载预设后学习率变成 0。原因:JSON 里学习率写成了无法解析的字符串,程序兜底置 0.0。对策:打开预设文件,把 learning_rate、text_encoder_lr、unet_lr 改成数字形式,如 3e-05。
📊 参数速查
| 参数 | 推荐值 | 适用场景 |
|---|---|---|
| learning_rate | 4e-7 | SDXL LoRA 训练,官方文档标注的标准值 |
| learning_rate | 1e-6 | Dreambooth 全参数微调 |
| learning_rate | 2e-4 | LLM Lite 相关训练 |
| lr_scheduler | cosine | 常规训练,config 示例默认值 |
| optimizer | AdamW8bit | 显存有限时的默认选择 |
数值出处见 docs/LoRA/top_level.md 与 docs/train_db_README-zh.md,GUI 默认值逻辑在 kohya_gui/class_basic_training.py 中。
学习率是训练里最敏感的参数,自动填充失效时先查文件、再查预设、最后手动填,顺序别乱。想看完整参数说明,可以读 docs/train_README.md;想理解预设如何加载,直接翻 kohya_gui/common_gui.py 里对 learning_rate 字段的转换逻辑。
【免费下载链接】kohya_ss项目地址: https://gitcode.com/GitHub_Trending/ko/kohya_ss
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考