From febaf863026bd014b7a14349336544fc109d0f57 Mon Sep 17 00:00:00 2001 From: Will Berman Date: Tue, 24 Jan 2023 08:37:34 -0800 Subject: [PATCH] [docs] [dreambooth] note random crop (#2085) * [docs] [dreambooth] note random crop documenting default random crop behavior --- examples/dreambooth/train_dreambooth.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 3df55937c4..2ecb8fb174 100644 --- a/examples/dreambooth/train_dreambooth.py +++ b/examples/dreambooth/train_dreambooth.py @@ -156,7 +156,10 @@ def parse_args(input_args=None): ), ) parser.add_argument( - "--center_crop", action="store_true", help="Whether to center crop images before resizing to resolution" + "--center_crop", + default=False, + action="store_true", + help="Whether to center crop images before resizing to resolution. If not set, images are randomly cropped.", ) parser.add_argument( "--train_text_encoder",