Skip to content

Commit accfa9e

Browse files
authored
fixing kernel restarting when transcribing (#14665)
* fixing kernel restarting when transcribing Signed-off-by: Weiqing Wang <weiqingw@nvidia.com> * fixing the same issue for tutorials/asr/ASR_with_NeMo.ipynb Signed-off-by: Weiqing Wang <weiqingw@nvidia.com> * remove the change caused by IDE Signed-off-by: Weiqing Wang <weiqingw@nvidia.com> --------- Signed-off-by: Weiqing Wang <weiqingw@nvidia.com>
1 parent 8201231 commit accfa9e

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tutorials/asr/ASR_with_NeMo.ipynb

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1732,6 +1732,8 @@
17321732
},
17331733
"outputs": [],
17341734
"source": [
1735+
"first_asr_model.cuda()\n",
1736+
"first_asr_model.eval()\n",
17351737
"audio = [os.path.join(data_dir, 'an4/wav/an4_clstk/mgah/cen2-mgah-b.wav'),\n",
17361738
" os.path.join(data_dir, 'an4/wav/an4_clstk/fmjd/cen7-fmjd-b.wav'),\n",
17371739
" os.path.join(data_dir, 'an4/wav/an4_clstk/fmjd/cen8-fmjd-b.wav'),\n",

tutorials/asr/ASR_with_Subword_Tokenization.ipynb

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -981,6 +981,8 @@
981981
"id": "O64yk8C4xvTG"
982982
},
983983
"source": [
984+
"first_asr_model.cuda()\n",
985+
"first_asr_model.eval()\n",
984986
"print(first_asr_model.transcribe(audio=[data_dir + '/an4/wav/an4_clstk/mgah/cen2-mgah-b.wav',\n",
985987
" data_dir + '/an4/wav/an4_clstk/fmjd/cen7-fmjd-b.wav',\n",
986988
" data_dir + '/an4/wav/an4_clstk/fmjd/cen8-fmjd-b.wav',\n",

0 commit comments

Comments
 (0)