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训练服务
sh run_finetune.sh
所有数据保存在auto_log/avg_step_time.txt文本中 auto_log/log/目录下存放各个shapes的数据