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

Remove redundant dataset import

This commit is contained in:
Eren Golge 2018-04-17 10:05:50 -07:00
parent 7361603095
commit f5e9c52abe
1 changed files with 29 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
train.py
View File

@ -23,7 +23,6 @@ from utils.generic_utils import (Progbar, remove_experiment_folder,
count_parameters, check_update, get_commit_hash) count_parameters, check_update, get_commit_hash)
from utils.model import get_param_size from utils.model import get_param_size
from utils.visual import plot_alignment, plot_spectrogram from utils.visual import plot_alignment, plot_spectrogram
from datasets.LJSpeech import LJSpeechDataset
from models.tacotron import Tacotron from models.tacotron import Tacotron
from layers.losses import L1LossMasked from layers.losses import L1LossMasked
@ -296,7 +295,7 @@ def main(args):
Dataset = getattr(mod, c.dataset+"Dataset") Dataset = getattr(mod, c.dataset+"Dataset")
# Setup the dataset # Setup the dataset
train_dataset = LJSpeechDataset(os.path.join(c.data_path, c.meta_file_train), train_dataset = Dataset(os.path.join(c.data_path, c.meta_file_train),
os.path.join(c.data_path, 'wavs'), os.path.join(c.data_path, 'wavs'),
c.r, c.r,
c.sample_rate, c.sample_rate,
@ -317,7 +316,7 @@ def main(args):
drop_last=False, num_workers=c.num_loader_workers, drop_last=False, num_workers=c.num_loader_workers,
pin_memory=True) pin_memory=True)
val_dataset = LJSpeechDataset(os.path.join(c.data_path, c.meta_file_val), val_dataset = Dataset(os.path.join(c.data_path, c.meta_file_val),
os.path.join(c.data_path, 'wavs'), os.path.join(c.data_path, 'wavs'),
c.r, c.r,
c.sample_rate, c.sample_rate,