From 9418f4f3dd23052c08810007aad1816896b39754 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Sat, 7 Oct 2023 01:59:18 +0000 Subject: [PATCH] fix --- submission_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/submission_runner.py b/submission_runner.py index 7b9e8423b..344394abe 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -273,7 +273,7 @@ def train_once( checkpoint_dir=log_dir) meta_file_name = os.path.join(log_dir, f'meta_data_{preemption_count}.json') logging.info(f'Saving meta data to {meta_file_name}.') - meta_data = logger_utils.get_meta_data(workload, rng_seed, preemption_count) + meta_data = logger_utils.get_meta_data(workload, rng_seed) logger_utils.write_json(meta_file_name, meta_data) flag_file_name = os.path.join(log_dir, f'flags_{preemption_count}.json') logging.info(f'Saving flags to {flag_file_name}.')