fix root path

This commit is contained in:
iclementine 2021-03-31 19:36:48 +08:00
parent dd73ee6611
commit a834e132b9
1 changed files with 2 additions and 1 deletions

View File

@ -10,7 +10,8 @@ from parakeet.data import batch_spec, batch_text_id
class VCTK(Dataset): class VCTK(Dataset):
def __init__(self, root): def __init__(self, root):
self.root = Path(root).expanduser() root = Path(root).expanduser()
self.root = root
record_path = self.root / "metadata.pickle" record_path = self.root / "metadata.pickle"
self.wav_root = root / "wav" self.wav_root = root / "wav"
self.mel_root = root / "wav" self.mel_root = root / "wav"