diff --git a/DeepSpeech.ipynb b/DeepSpeech.ipynb index ca22ddea..b32d4c39 100644 --- a/DeepSpeech.ipynb +++ b/DeepSpeech.ipynb @@ -90,8 +90,8 @@ "from util.text import sparse_tensor_value_to_texts, wers\n", "from tensorflow.python.ops import ctc_ops\n", "\n", - "ds_importer = os.environ.get('ds_importer', 'ted_lium')\n", - "ds_dataset_path = os.environ.get('ds_dataset_path', './data/ted')\n", + "ds_importer = os.environ.get('ds_importer', 'ted')\n", + "ds_dataset_path = os.environ.get('ds_dataset_path', os.path.join('./data', ds_importer))\n", "\n", "import importlib\n", "ds_importer_module = importlib.import_module('util.importers.%s' % ds_importer)\n", diff --git a/util/importers/ted_lium.py b/util/importers/ted.py similarity index 100% rename from util/importers/ted_lium.py rename to util/importers/ted.py