使用 sd-webui 快速训练 embedding

sd-webui embedding 训练

启动 sd-webui

./webui.sh --opt-split-attention-v1

启动后，我本机使用默认端口号 7860 则直接访问 http://127.0.0.1:7860/

建立 embedding

起个名字如：fist_of_north_star
词元向量：3

预处理图片

抠图

把干扰识别的部分抠下去，只留要识别的部分。

标注提示词

webui提供工具进行图片内容提取，也可以不用他的，自己写提示词，提示词保存为图片文件名一样的txt文件。

列如无视以上过程生成的结果，强制将提示词进行统一更改：

# 在结果目录下
echo  fist_of_north_star fist north star  > `ls *.txt`

训练配置

学习速率可参考以下配置，速度约低，误差梯度下降越慢

0.05:10, 0.02:20, 0.01:60, 0.005:200, 0.002:500, 0.001:3000, 0.0005

数据集目录选带图片和提示词(txt文件)的目录

批次和累加步数相乘要大于等于图片数量，批次根据显存大小试验调整

多少步保存图像和模型副本，可以随意选，为的是可以在训练过程中提前看到当前训练的结果。

最后触发训练 embedding 指令。训练结果的副本会随着以上的设置被保存下来。

训练过程中注意CPU和GPU散热。

如果一次跑不完需要中途停下来，可以触发中止指令。误差越低，证明训练结果越好。