What steps, learning rates, class-image counts and VRAM figures do the DreamBooth and textual-inversion papers and the diffusers v0.41.0 docs state, and where do they differ?
As stated: DreamBooth paper ~1000 iterations, prior-loss weight lambda=1, learning rate 1e-5 (Imagen) or 5e-6 (Stable Diffusion), 3-5 images, ~1000 class samples (fewer possible), about 5 minutes on one A100 for Stable Diffusion. diffusers examples/dreambooth README: 5e-6 with 400 steps (no prior), 5e-6 with 800 steps and 200 class images (prior-preservation), 2e-6 for the 12 GB variant, LoRA DreamBooth 1e-4 and 500 steps, with the text encoder needing at least 24 GB. Textual inversion paper: batch size 4, base lr 0.005 scaled to 0.04, 5,000 steps on 2 V100s, about two hours per concept; diffusers README: 5.0e-04 with --scale_lr and 3000 steps. Hugging Face's own blog experiments report 400 steps at 2e-6 for objects and ~1200 steps at 1e-6 for faces. The values differ by source and model, and are used as stated; none is a universal setting.
Structured fields
{
"citations": [
{
"url": "https://arxiv.org/pdf/2208.12242v2",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://arxiv.org/html/2208.01618v1",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/dreambooth/train_dreambooth.py",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/dreambooth/README.md",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/textual_inversion/README.md",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/blog/blob/cab0e22b682925b758308ea1f51456491f9fff34/dreambooth.md",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/docs/source/en/training/dreambooth.md",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
},
{
"url": "https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/docs/source/en/training/lora.md",
"note": "producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;DR and Learning Rate Impact"
}
]
}Evidence claims
- [sourced_fact] As stated: DreamBooth paper ~1000 iterations, prior-loss weight lambda=1, learning rate 1e-5 (Imagen) or 5e-6 (Stable Diffusion), 3-5 images, ~1000 class samples (fewer possible), about 5 minutes on one A100 for Stable Diffusion. diffusers examples/dreambooth README: 5e-6 with 400 steps (no prior), 5e-6 with 800 steps and 200 class images (prior-preservation), 2e-6 for the 12 GB variant, LoRA DreamBooth 1e-4 and 500 steps, with the text encoder needing at least 24 GB. Textual inversion paper: batch size 4, base lr 0.005 scaled to 0.04, 5,000 steps on 2 V100s, about two hours per concept; diffusers README: 5.0e-04 with --scale_lr and 3000 steps. Hugging Face's own blog experiments report 400 steps at 2e-6 for objects and ~1200 steps at 1e-6 for faces. The values differ by source and model, and are used as stated; none is a universal setting. · locator: producer work copy work/dreambooth.pdf (sha256 equals pin; local read) Sec. 3.3 final paragraph; work/textinv.html (equals pin) implementation details; limitations; raw GET at 086bf957 (bytes equal the work copy main_snapshot/db_README.md and the pin) L82-L353; raw GET at 086bf957 (equals pin) L70-L82; raw GET at 086bf957 (equals pin) L155; work/train_db.py (equals pin) argument defaults L293-L297, L391; work/train_db_lora.py (equals pin) --learning_rate default; work/dbblog.md (equals pin) TL;D source
Rights
| component | license | hosting | redistribution | modification | training | Evidence and notes |
|---|---|---|---|---|---|---|
| documentation | unknown | permitted | permitted | unknown | unknown | Applies to this original summary only, not to the cited sources. |
Linked records
- cites → https://arxiv.org/html/2208.01618v1
- cites → https://arxiv.org/pdf/2208.12242v2
- cites → https://github.com/huggingface/blog/blob/cab0e22b682925b758308ea1f51456491f9fff34/dreambooth.md
- cites → https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/docs/source/en/training/dreambooth.md
- cites → https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/docs/source/en/training/lora.md
- cites → https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/dreambooth/README.md
- cites → https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/dreambooth/train_dreambooth.py
- cites → https://github.com/huggingface/diffusers/blob/086bf9578c0f4acbc66e48cd1e7cc26befd9e10f/examples/textual_inversion/README.md