change bsz back

mlcommons · Sep 26, 2023 · ddbba17 · ddbba17
1 parent 372f536
commit ddbba17
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/workload.py
@@ -55,7 +55,7 @@ def num_test_examples(self) -> int:
 
   @property
   def eval_batch_size(self) -> int:
-    return 32
+    return 256
 
   @property
   def train_mean(self):

diff --git a/reference_algorithms/target_setting_algorithms/get_batch_size.py b/reference_algorithms/target_setting_algorithms/get_batch_size.py
@@ -12,7 +12,7 @@ def get_batch_size(workload_name):
   elif workload_name == 'imagenet_vit':
     return 1024
   elif workload_name == 'librispeech_conformer':
-    return 32
+    return 256
   elif workload_name == 'librispeech_deepspeech':
     return 256
   elif workload_name == 'ogbg':

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_adamw.pickle", "wb") as f:
+      with open(f"/experiment_runs/snapshot_{global_step}_full_bsz_adamw_reduced_alloc_size.pickle", "wb") as f:
         dump(s, f)
   metrics = {'eval_results': eval_results, 'global_step': global_step}