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
2022-10-20 13:44:37 +02:00 · 2022-10-20 13:44:37 +02:00 · ce7d96681c
parent db19a9d9d7
commit ce7d96681c
1 changed files with 1 additions and 0 deletions
--- a/examples/dreambooth/README.md
+++ b/examples/dreambooth/README.md
@ -151,6 +151,7 @@ accelerate launch train_dreambooth.py \
  --class_prompt="a photo of dog" \
  --resolution=512 \
  --train_batch_size=1 \
+  --sample_batch_size=1 \
  --gradient_accumulation_steps=1 --gradient_checkpointing \
  --learning_rate=5e-6 \
  --lr_scheduler="constant" \