This commit is contained in:
Linaqruf
2023-04-22 16:17:22 +07:00

View File

@@ -2754,7 +2754,9 @@ def prepare_accelerator(args: argparse.Namespace):
log_prefix = "" if args.log_prefix is None else args.log_prefix 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()) 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 log_with = args.log_with
if log_with in ["tensorboard", "all"]: if log_with in ["tensorboard", "all"]:
if logging_dir is None: if logging_dir is None: