More notebook work

notebooks/easy_transfer_learning.ipynb

@@ -147,7 +147,7 @@
     "    alphabet_config_path=\"russian/alphabet.txt\",\n",
     "    train_files=[\"russian/ru.csv\"],\n",
     "    dev_files=[\"russian/ru.csv\"],\n",
-    "    epochs=200,\n",
+    "    epochs=100,\n",
     "    load_cudnn=True,\n",
     ")"
    ]

notebooks/train_with_common_voice.ipynb

@@ -92,21 +92,32 @@
       ],
       "id": "be5fe49c"
     },
-    {
-      "cell_type": "code",
-      "metadata": {
-        "scrolled": true,
-        "id": "53945462"
-      },
-      "source": [
-        "### Download pre-formatted Common Voice data\n",
-        "! wget https://coqui-ai-public-data.s3.amazonaws.com/cv/7.0/kk-data.tar\n",
-        "! tar -xf kk-data.tar"
-      ],
-      "id": "53945462",
-      "execution_count": null,
-      "outputs": []
-    },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "608d203f",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "### Download pre-formatted Common Voice data\n",
+    "import os\n",
+    "import tarfile\n",
+    "from coqui_stt_training.util.downloader import maybe_download\n",
+    "\n",
+    "def download_preformatted_data():\n",
+    "    if not os.path.exists(\"data/sr-data\"):\n",
+    "        maybe_download(\"sr-data.tar\", \"data/\", \"https://coqui-ai-public-data.s3.amazonaws.com/cv/7.0/sr-data.tar\")\n",
+    "        print('\\nNo extracted data found. Extracting now...')\n",
+    "        tar = tarfile.open(\"data/sr-data.tar\", mode=\"r:\")\n",
+    "        tar.extractall(\"data/\")\n",
+    "        tar.close()\n",
+    "    else:\n",
+    "        print('Found \"data/sr-data\" - not extracting.')\n",
+    "\n",
+    "# Download + extract Common Voice data\n",
+    "download_preformatted_data()"
+   ]
+  },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -123,8 +134,8 @@
         "id": "fa2aec77"
       },
       "source": [
-        "! ls kk-data\n",
-        "! wc -l kk-data/*.csv"
+        "! ls data/sr-data\n",
+        "! wc -l data/sr-data/*.csv"
       ],
       "id": "fa2aec77",
       "execution_count": null,
@@ -155,9 +166,9 @@
         "from coqui_stt_training.util.config import initialize_globals_from_args\n",
         "\n",
         "initialize_globals_from_args(\n",
-        "    train_files=[\"kk-data/train.csv\"],\n",
-        "    dev_files=[\"kk-data/dev.csv\"],\n",
-        "    test_files=[\"kk-data/test.csv\"],\n",
+        "    train_files=[\"data/sr-data/train.csv\"],\n",
+        "    dev_files=[\"data/sr-data/dev.csv\"],\n",
+        "    test_files=[\"data/sr-data/test.csv\"],\n",
         "    load_train=\"init\",\n",
         "    n_hidden=200,\n",
         "    epochs=1,\n",

notebooks/train_your_first_coqui_STT_model.ipynb

@@ -151,7 +151,7 @@
     "    test_files=[\"english/ldc93s1.csv\"],\n",
     "    load_train=\"init\",\n",
     "    n_hidden=100,\n",
-    "    epochs=200,\n",
+    "    epochs=100,\n",
     ")"
    ]
   },