annotate

mlcommons · Sep 25, 2023 · 372f536 · 372f536
1 parent b8ca663
commit 372f536
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}