Update train_dreambooth.py

2023-06-08 08:27:06 +04:00 · 2023-06-08 08:27:06 +04:00 · b320b59ff6
parent 2933b6b20b
commit b320b59ff6
1 changed files with 2 additions and 2 deletions
--- a/examples/dreambooth/train_dreambooth.py
+++ b/examples/dreambooth/train_dreambooth.py
@ -426,7 +426,7 @@ def main(args):
        gradient_accumulation_steps=args.gradient_accumulation_steps,
        mixed_precision=args.mixed_precision,
        log_with="tensorboard",
-        logging_dir=logging_dir,
+        project_dir=logging_dir,
    )

    logging.basicConfig(