Hugging Face Blog·· 2022-11-07精选AI 评分62
用 Diffusers 和 Dreambooth 训练 Stable Diffusion 的实验与调参建议
Training Stable Diffusion with Dreambooth using Diffusers
AI 导读
Hugging Face 发布用 Diffusers 的 Dreambooth 脚本微调 Stable Diffusion 的实验总结,给出推荐设置:低学习率配合逐步增加训练步数,人脸需要 800-1200 步、batch size 2、LR 1e-6,并建议使用 prior preservation 避免过拟合。
推荐理由
基于大量实验给出 Dreambooth 微调 Stable Diffusion 的超参数建议,可直接用于复现和调参。
来源:Hugging Face Blog · huggingface.co