aboutsummaryrefslogtreecommitdiff
path: root/train.py
diff options
context:
space:
mode:
Diffstat (limited to 'train.py')
-rw-r--r--train.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/train.py b/train.py
new file mode 100644
index 0000000..ebeb4d4
--- /dev/null
+++ b/train.py
@@ -0,0 +1,11 @@
+#!/bin/env python3
+
+import torchaudio
+
+# download the datasets
+train_dataset = torchaudio.datasets.LIBRISPEECH(
+ "./data", url="train-clean-100", download=True
+)
+test_dataset = torchaudio.datasets.LIBRISPEECH(
+ "./data", url="test-clean", download=True
+)