From b320b59ff652c0c3c7c8304dbdb09f254dc8bd63 Mon Sep 17 00:00:00 2001 From: natkaha Date: Thu, 8 Jun 2023 08:27:06 +0400 Subject: [PATCH] Update train_dreambooth.py --- examples/dreambooth/train_dreambooth.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py index 684d9f57..b4880bbc 100644 --- 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( @@ -866,4 +866,4 @@ def main(args): if __name__ == "__main__": args = parse_args() - main(args) \ No newline at end of file + main(args)