From 33cb4872a78c9681aa3c7299280e22a56ca613ba Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Wed, 29 Nov 2023 07:46:44 +0000 Subject: [PATCH] fix --- .../exp2023_04_25_rf/conformer_import_moh_att_2023_06_30.py | 3 ++- 1 file changed, 2 insertions(+), 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 index 5a6e48e4e..7bb0254b7 100644 --- 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 )