diff options
Diffstat (limited to 'dreambooth.py')
-rw-r--r-- | dreambooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/dreambooth.py b/dreambooth.py index bc7a472..7c0d32c 100644 --- a/dreambooth.py +++ b/dreambooth.py | |||
@@ -65,7 +65,7 @@ def parse_args(): | |||
65 | parser.add_argument( | 65 | parser.add_argument( |
66 | "--output_dir", | 66 | "--output_dir", |
67 | type=str, | 67 | type=str, |
68 | default="dreambooth-model", | 68 | default="output/dreambooth", |
69 | help="The output directory where the model predictions and checkpoints will be written.", | 69 | help="The output directory where the model predictions and checkpoints will be written.", |
70 | ) | 70 | ) |
71 | parser.add_argument( | 71 | parser.add_argument( |