Unverified Commit ce7d9668 authored by Hanusz Leszek's avatar Hanusz Leszek Committed by GitHub
Browse files

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
parent db19a9d9
...@@ -151,6 +151,7 @@ accelerate launch train_dreambooth.py \ ...@@ -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" \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment