fix root path

2021-03-31 19:36:48 +08:00 · 2021-03-31 19:36:48 +08:00 · a834e132b9
parent dd73ee6611
commit a834e132b9
1 changed files with 2 additions and 1 deletions
--- a/examples/tacotron2_vctk/vctk.py
+++ b/examples/tacotron2_vctk/vctk.py
@ -10,7 +10,8 @@ from parakeet.data import batch_spec, batch_text_id

 class VCTK(Dataset):
    def __init__(self, root):
-        self.root = Path(root).expanduser()
+        root = Path(root).expanduser()
+        self.root = root
        record_path = self.root / "metadata.pickle"
        self.wav_root = root / "wav"
        self.mel_root = root / "wav"