fix no logging command line arg

This commit is contained in:
saltacc
2023-04-22 01:26:31 -07:00
committed by GitHub
parent f256660780
commit dc37fd2ff6

View File

@@ -2761,6 +2761,8 @@ 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,