From eb9a166d06bbb07e13b7b7abdc3a63d034b7f359 Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Thu, 22 Sep 2022 13:29:02 -0700 Subject: [PATCH 1/2] Fix ASR issues Signed-off-by: smajumdar --- examples/asr/conf/marblenet/marblenet_3x2x64.yaml | 2 +- tutorials/asr/ASR_with_NeMo.ipynb | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/asr/conf/marblenet/marblenet_3x2x64.yaml b/examples/asr/conf/marblenet/marblenet_3x2x64.yaml index f9b3f26d114c..b19688d902b8 100644 --- a/examples/asr/conf/marblenet/marblenet_3x2x64.yaml +++ b/examples/asr/conf/marblenet/marblenet_3x2x64.yaml @@ -167,7 +167,7 @@ trainer: max_epochs: 150 max_steps: -1 # computed at runtime if not set num_nodes: 1 - accelerator: gpu + accelerator: cuda strategy: ddp accumulate_grad_batches: 1 enable_checkpointing: False # Provided by exp_manager diff --git a/tutorials/asr/ASR_with_NeMo.ipynb b/tutorials/asr/ASR_with_NeMo.ipynb index 99489c50becd..7ce61417f3e9 100644 --- a/tutorials/asr/ASR_with_NeMo.ipynb +++ b/tutorials/asr/ASR_with_NeMo.ipynb @@ -161,6 +161,7 @@ "id": "gAhsmi6HlRzh" }, "source": [ + "import os\n", "# This is where the an4/ directory will be placed.\n", "# Change this if you don't want the data to be extracted in the current directory.\n", "data_dir = '.'\n", From dfd30965395834661d1a08fdb6ec706a5761fc7a Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Thu, 22 Sep 2022 23:01:31 -0700 Subject: [PATCH 2/2] Revert fix Signed-off-by: smajumdar --- examples/asr/conf/marblenet/marblenet_3x2x64.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/asr/conf/marblenet/marblenet_3x2x64.yaml b/examples/asr/conf/marblenet/marblenet_3x2x64.yaml index b19688d902b8..f9b3f26d114c 100644 --- a/examples/asr/conf/marblenet/marblenet_3x2x64.yaml +++ b/examples/asr/conf/marblenet/marblenet_3x2x64.yaml @@ -167,7 +167,7 @@ trainer: max_epochs: 150 max_steps: -1 # computed at runtime if not set num_nodes: 1 - accelerator: cuda + accelerator: gpu strategy: ddp accumulate_grad_batches: 1 enable_checkpointing: False # Provided by exp_manager