From faf775e1ca24d634e4021253788c23b279e7cb40 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Mon, 25 Sep 2023 19:57:12 +0000 Subject: [PATCH] debug --- submission_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/submission_runner.py b/submission_runner.py index 999b978c0..83130d4a9 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -424,7 +424,7 @@ def train_once( train_state['last_step_end_time'] = get_time() if global_step == 2: s = torch.cuda.memory._snapshot() - with open(f"/experiment_runs/snapshot_{global_step}_half_bsz.pickle", "wb") as f: + with open(f"/experiment_runs/snapshot_{global_step}_full_bsz.pickle", "wb") as f: dump(s, f) metrics = {'eval_results': eval_results, 'global_step': global_step}