分享
Dreambooth 官方炼丹教程万字详解-Epochs\Batch size\学习率 等超参数调优 (一)
输入“/”快速插入内容
Dreambooth 官方炼丹教程万字详解-Epochs\Batch size\学习率 等超参数调优 (一)
用户8537
用户8537
2023年9月5日修改
20%
📣
作者:设计师忠忠
Stable Diffusion 炼丹阁
.
AIGC炼丹师聚集地,专注于将Stable Diffusion的图像生成与模型训练能力应用于落地工作。
80%
原创
设计师忠忠 Stable Diffusion 炼丹阁
2023-04-09 18:16
发表于广东
原文链接地址:
https://mp.weixin.qq.com/s/8ECZ5xaUF20AqMU3jb2Zqg
前言
Dreambooth 官方 Github 的 Readme 文档上有着非常详细的文字说明与视频教程,是学习 Dreambooth 炼丹的最佳资料之一,讲解了包括
训练集训练轮数Epochs 、同时处理的图片数 Batch Size 、学习率 Learning Rate
等核心超参数在内的各种参数设置与调整建议。
接下来我将结合自己的理解,靠着 GPT4、NewBing 的帮助和大家一起修炼这份炼丹术进阶教程,踏上这炼丹修仙长生之路!
Dreambooth Extension for Stable-Diffusion-WebUI
这是Shivam Shriao的Diffusers Repo的一个正在进行中的移植版本,该版本是基于Huggingface Diffusers Repo的默认版本进行修改的,以
在低VRAM GPU(低显存显卡)上获得更好的性能
。此外,还借鉴了BMaltais的Koyha SS的部分内容。它还添加了其他一些功能,包括同时训练多个概念,以及(即将推出)图像修复训练。
样式说明:灰色正文主要来自官方文档翻译,有些地方不通顺,我会做一定程度的更改。黑色正文则是我自己的理解和总结,如有错漏,欢迎斧正。
Installation 安装
要安装,请在SD Web UI中转到“Extensions(扩展)”选项卡,选择“Available(可用)”子选项卡,选择“Load from:(从...加载)”以加载扩展列表,最后在Dreambooth条目旁边点击“install(安装)”。
安装完成后,您必须完全重新启动Stable-Diffusion WebUI。重新加载UI将无法安装所需的依赖项。
我们还需要更新版本的diffusers,因为SD-WebUI使用的是0.3.0版本,而DB训练需要>= 0.10.0版本。没有正确的diffusers版本会导致“UNet2DConditionModel”对象没有属性“enable_gradient_checkpointing”的错误消息,以及安全检查器警告。
如果在安装后遇到依赖项问题,请查看以下内容
为了强制sd-web-ui只安装一套依赖项并解决许多安装问题,我们可以指定命令行参数:
•
代码块
Plain Text
set/exportREQS_FILE=.\extensions\sd_dreambooth_extension\requirements.txt
请参考下面的适当脚本,以添加额外的标志来安装依赖项:
https://github.com/d8ahazard/sd_dreambooth_extension/blob/main/webui-user-dreambooth.bat
https://github.com/d8ahazard/sd_dreambooth_extension/blob/main/webui-user-dreambooth.sh
最后,如果您希望完全跳过Dreambooth的“本地”安装过程,您可以设置以下环境标志: