跳到正文
原文
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