fix

rwth-i6 · Nov 29, 2023 · 33cb487 · 33cb487
1 parent 1906043
commit 33cb487
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/users/zeyer/experiments/exp2023_04_25_rf/conformer_import_moh_att_2023_06_30.py b/users/zeyer/experiments/exp2023_04_25_rf/conformer_import_moh_att_2023_06_30.py
@@ -47,13 +47,14 @@ def sis_run_with_prefix(prefix_name: Optional[str] = None):
     train_exp("from-scratch-train", config, gpu_mem=11)
 
     train_exp(
-        "v6-11gb-bs15k-accgrad4-mgpu2",
+        "v6-11gb-f32-bs15k-accgrad4-mgpu2",
         config_24gb_v6,
         config_updates={
             "batch_size": 15_000 * _batch_size_factor,
             "accum_grad_multiple_step": 4,
             "torch_distributed": {},  # multi-GPU
         },
+        config_deletes=["torch_amp"],  # f32
         gpu_mem=11,
         num_processes=2,  # multi-GPU
     )