Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DOC Dreambooth Add --sample_batch_size=1 to the 8 GB dreambooth example script (#829) 

Add --sample_batch_size=1 to the 8 GB dreambooth script
This commit is contained in:
Hanusz Leszek 2022-10-20 13:44:37 +02:00 committed by GitHub
parent db19a9d9d7
commit ce7d96681c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/dreambooth/README.md
View File

@ -151,6 +151,7 @@ accelerate launch train_dreambooth.py \
--class_prompt="a photo of dog" \ --class_prompt="a photo of dog" \
--resolution=512 \ --resolution=512 \
--train_batch_size=1 \ --train_batch_size=1 \
--sample_batch_size=1 \
--gradient_accumulation_steps=1 --gradient_checkpointing \ --gradient_accumulation_steps=1 --gradient_checkpointing \
--learning_rate=5e-6 \ --learning_rate=5e-6 \
--lr_scheduler="constant" \ --lr_scheduler="constant" \