From 372f536d2f89202124f54021d9006b3d2a2d6ed1 Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Mon, 25 Sep 2023 22:17:51 +0000 Subject: [PATCH] annotate --- submission_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/submission_runner.py b/submission_runner.py index 224dfda28..5519399a8 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -426,7 +426,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}_full_bsz.pickle", "wb") as f: + with open(f"/experiment_runs/snapshot_{global_step}_full_bsz_adamw.pickle", "wb") as f: dump(s, f) metrics = {'eval_results': eval_results, 'global_step': global_step}