更新时间:2024-09-10 GMT+08:00
分享

Finetune训练

本章节介绍SDXL&SD 1.5模型的Finetune训练过程。Finetune是指在已经训练好的模型基础上,使用新的数据集进行微调(fine-tuning)以优化模型性能。修改数据集路径、模型路径。数据集路径格式为/datasets/pokemon-dataset/image_0.png,脚本里写到pokemon-dataset路径即可。

kohya_finetune.toml文件里数据集路径更改为pokemon-dataset路径。

cd koyha_ss
cp run_* sd-scripts
cp kohya_finetune.toml sd-scripts
cp meta_cap.json sd-scripts
cd sd-scripts
vim run_finetune.sh
vim kohya_finetune.toml
python finetune/make_captions.py {数据集路径pokemon-dataset路径}

启动SD1.5 Finetune训练服务

使用ma-user用户执行如下命令运行训练脚本。
sh run_finetune.sh

所有数据保存在auto_log/avg_step_time.txt文本中 auto_log/log/目录下存放各个shapes的数据

相关文档