Add PyTorchIcefall to estimator init

Trusted-AI · Jan 8, 2024 · 2363a7b · 2363a7b
1 parent e3aac52
commit 2363a7b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 2 deletions.
diff --git a/art/estimators/speech_recognition/__init__.py b/art/estimators/speech_recognition/__init__.py
@@ -5,4 +5,5 @@
 
 from art.estimators.speech_recognition.pytorch_deep_speech import PyTorchDeepSpeech
 from art.estimators.speech_recognition.pytorch_espresso import PyTorchEspresso
+from art.estimators.speech_recognition.pytorch_icefall import PyTorchIcefall
 from art.estimators.speech_recognition.tensorflow_lingvo import TensorFlowLingvoASR
diff --git a/notebooks/asr_deepspeech_examples.ipynb b/notebooks/asr_deepspeech_examples.ipynb
@@ -22,7 +22,27 @@
    "cell_type": "code",
    "execution_count": 1,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/home/xli257/miniconda3/envs/slu_icefall/lib/python3.9/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
+     ]
+    },
+    {
+     "ename": "ModuleNotFoundError",
+     "evalue": "No module named 'deepspeech_pytorch'",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m                       Traceback (most recent call last)",
+      "\u001b[1;32m/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb Cell 3\u001b[0m line \u001b[0;36m7\n\u001b[1;32m      <a href='vscode-notebook-cell://ssh-remote%2Bb17/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb#W2sdnNjb2RlLXJlbW90ZQ%3D%3D?line=4'>5</a>\u001b[0m \u001b[39mimport\u001b[39;00m \u001b[39mIPython\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mdisplay\u001b[39;00m \u001b[39mas\u001b[39;00m \u001b[39mipd\u001b[39;00m\n\u001b[1;32m      <a href='vscode-notebook-cell://ssh-remote%2Bb17/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb#W2sdnNjb2RlLXJlbW90ZQ%3D%3D?line=5'>6</a>\u001b[0m \u001b[39mimport\u001b[39;00m \u001b[39mmatplotlib\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mpyplot\u001b[39;00m \u001b[39mas\u001b[39;00m \u001b[39mplt\u001b[39;00m\n\u001b[0;32m----> <a href='vscode-notebook-cell://ssh-remote%2Bb17/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb#W2sdnNjb2RlLXJlbW90ZQ%3D%3D?line=6'>7</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mdeepspeech_pytorch\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mloader\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mdata_loader\u001b[39;00m \u001b[39mimport\u001b[39;00m load_audio\n\u001b[1;32m      <a href='vscode-notebook-cell://ssh-remote%2Bb17/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb#W2sdnNjb2RlLXJlbW90ZQ%3D%3D?line=8'>9</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mart\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mestimators\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mspeech_recognition\u001b[39;00m \u001b[39mimport\u001b[39;00m PyTorchDeepSpeech\n\u001b[1;32m     <a href='vscode-notebook-cell://ssh-remote%2Bb17/home/xli257/slu/adversarial-robustness-toolbox/notebooks/asr_deepspeech_examples.ipynb#W2sdnNjb2RlLXJlbW90ZQ%3D%3D?line=9'>10</a>\u001b[0m \u001b[39mfrom\u001b[39;00m \u001b[39mart\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mattacks\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mevasion\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mimperceptible_asr\u001b[39;00m\u001b[39m.\u001b[39;00m\u001b[39mimperceptible_asr_pytorch\u001b[39;00m \u001b[39mimport\u001b[39;00m ImperceptibleASRPyTorch\n",
+      "\u001b[0;31mModuleNotFoundError\u001b[0m: No module named 'deepspeech_pytorch'"
+     ]
+    }
+   ],
    "source": [
     "import os\n",
     "\n",
@@ -678,7 +698,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.10"
+   "version": "3.9.18"
   }
  },
  "nbformat": 4,