From c1dac636292a2ea3de75744c9900a0a5eaaf3d83 Mon Sep 17 00:00:00 2001 From: Alex Fang Date: Mon, 20 May 2024 18:39:55 -0700 Subject: [PATCH] args.failed_checkpoint_path makedir --- open_lm/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/open_lm/main.py b/open_lm/main.py index 8efe3b5c..7c80f558 100644 --- a/open_lm/main.py +++ b/open_lm/main.py @@ -382,7 +382,7 @@ def main(args): args.failed_checkpoint_path = os.path.join(log_base_path, "checkpoints_failed") if is_master(args): args.tensorboard_path = os.path.join(log_base_path, "tensorboard") if args.tensorboard else "" - for dirname in [args.tensorboard_path, args.checkpoint_path]: + for dirname in [args.tensorboard_path, args.checkpoint_path, args.failed_checkpoint_path]: if dirname: os.makedirs(dirname, exist_ok=True) else: