Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Run ready

This commit is contained in:
Eren Golge 2018-03-02 08:01:04 -08:00
parent d7592621d1
commit 69e158c81a
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.json
View File

@ -25,5 +25,5 @@
"checkpoint": false, "checkpoint": false,
"save_step": 69, "save_step": 69,
"data_path": "/run/shm/erogol/LJSpeech-1.0", "data_path": "/run/shm/erogol/LJSpeech-1.0",
"output_path": "result", "output_path": "result"
} }

1
datasets/LJSpeech.py
View File

@ -18,7 +18,6 @@ class LJSpeechDataset(Dataset):
with open(csv_file, "r") as f: with open(csv_file, "r") as f:
self.frames = [line.split('|') for line in f] self.frames = [line.split('|') for line in f]
self.frames = self.frames[:256]
self.root_dir = root_dir self.root_dir = root_dir
self.outputs_per_step = outputs_per_step self.outputs_per_step = outputs_per_step
self.sample_rate = sample_rate self.sample_rate = sample_rate

2
train.py
View File

@ -285,6 +285,7 @@ def evaluate(model, criterion, data_loader, current_step):
def main(args): def main(args):
# Setup the dataset
train_dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata_train.csv'), train_dataset = LJSpeechDataset(os.path.join(c.data_path, 'metadata_train.csv'),
os.path.join(c.data_path, 'wavs'), os.path.join(c.data_path, 'wavs'),
c.r, c.r,
@ -325,6 +326,7 @@ def main(args):
drop_last=True, num_workers= 4, drop_last=True, num_workers= 4,
pin_memory=True) pin_memory=True)
model = Tacotron(c.embedding_size,
c.hidden_size, c.hidden_size,
c.num_mels, c.num_mels,
c.num_freq, c.num_freq,