From 66edc5af7bd1f81e084ef6cff3fa6de16ffeef8a Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sat, 22 Apr 2023 18:05:19 +0900 Subject: [PATCH] invert condition for checking log_with --- library/train_util.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/library/train_util.py b/library/train_util.py index eb46fbb0..83d0d6bb 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -2748,7 +2748,9 @@ def prepare_accelerator(args: argparse.Namespace): log_prefix = "" if args.log_prefix is None else args.log_prefix logging_dir = args.logging_dir + "/" + log_prefix + time.strftime("%Y%m%d%H%M%S", time.localtime()) - if args.log_with is not None: + if args.log_with is None: + log_with = None + else: log_with = args.log_with if log_with in ["tensorboard", "all"]: if logging_dir is None: @@ -2761,8 +2763,6 @@ def prepare_accelerator(args: argparse.Namespace): if logging_dir is not None: os.makedirs(logging_dir, exist_ok=True) os.environ["WANDB_DIR"] = logging_dir - else: - log_with = None accelerator = Accelerator( gradient_accumulation_steps=args.gradient_accumulation_steps,