训练lora小模型

一，安装部署本地训练环境

1，下载源码

git clone --recurse-submodules https://github.com/Akegarasu/lora-scripts

右键 install.ps1，powershell打开，会自动安装需要的环境

2，下载模型

准备对应lora风格的模型，比如你要训练动漫人物，就找一个二次元风格的大模型，当作底膜训练

国内sd大模型网站：https://www.liblib.art
国外sd大模型网站（科学）：https://civitai.com

二，准备数据

1，准备图片

准备30-100张图片，图片风格就是要训练的lora模型的风格

2，标注图片

先把图片裁剪到相同尺寸然后在打标签

本文使用sd自带的打标签功能

如果图片不够清晰，也可以使用sd的高清化模块，先2倍高清化图片

三，修改配置

1，修改文件名

把处理好的图片放在train目录下
格式为：

10_cat 文件最前面的 10 表示每张图片训练的轮数，可按需调整
train
----cat
------10_cat

2，修改配置文件 ，install.ps1

如果是ui界面训练可以跳过这一步

需要修改的地方
1，$pretrained_model = "./sd-models/model.ckpt"，对应这底模的路径
2，$train_data_dir = "./train/aki"，训练集目录
3，$resolution = "512,512"，图片的宽高
4，$batch_size = 1，批次大小，关乎着模型的收敛速度（对于深度学习不熟悉的，保持默认即可）
5，$max_train_epoches = 10，训练步数
6，$save_every_n_epochs = 2，每隔几轮保存一次模型（可用于不同步数模型的测试，对比效果）
7，$output_name = "aki"，模型输出名称
8，$save_model_as = "safetensors"，保存模型的类型


配置文件:
# LoRA train script by @Akegarasu

# Train data path | 设置训练用模型、图片
$pretrained_model = "./sd-models/model.ckpt" # base model