fix reinfo train data

NTT123 · May 6, 2021 · b86fc00 · b86fc00
1 parent a93daad
commit b86fc00
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,3 +1,4 @@
+train_data/
 test_data/
 assets/reinfo/
 # IDE files

diff --git a/README.md b/README.md
@@ -27,7 +27,7 @@ Download reinfo dataset
 -----------------------
 
 ```sh
-bash ./scripts/download_reinfo_dataset.sh
+bash ./scripts/download_aligned_reinfo_dataset.sh
 ```
 
 

diff --git a/scripts/download_aligned_reinfo_dataset.sh b/scripts/download_aligned_reinfo_dataset.sh
@@ -0,0 +1,7 @@
+data_root="./train_data" # modify this
+pushd .
+mkdir -p $data_root
+cd $data_root
+gdown --id 1Pe-5lKT_lZsliv2WxQDai2mjhI9ZMFlj -O reinfo.zip
+unzip reinfo.zip 
+popd
diff --git a/setup.py b/setup.py
@@ -1,6 +1,6 @@
 from setuptools import setup
 
-__version__ = '0.1'
+__version__ = '0.1.1'
 url = 'https://github.com/ntt123/vietTTS'
 
 install_requires = ['tabulate', 'optax', 'jax', 'jaxlib', 'einops', 'librosa',

diff --git a/vietTTS/nat/config.py b/vietTTS/nat/config.py
@@ -33,7 +33,7 @@ class FLAGS(Namespace):
   # ckpt
   ckpt_dir = Path('assets/reinfo/nat')
   data_dir = Path('assets/reinfo/data')
-  data_dir = Path('test_data')
+  data_dir = Path('train_data')
 
 
 class DurationInput(NamedTuple):
-Original file line number
+Diff line change
@@ Expand Up / @@ -27,7 +27,7 @@ Download reinfo dataset @@
     -----------------------
     ```sh
-    bash ./scripts/download_reinfo_dataset.sh
+    bash ./scripts/download_aligned_reinfo_dataset.sh
     ```
@@ Expand Down @@