mozilla
/
TTS
зеркало из https://github.com/mozilla/TTS.git
1
0
Форкнуть 0
Код Задачи Пакеты Проекты Релизы Вики Активность

Logging bug fix for evaly

This commit is contained in:
Eren G 2018-07-19 16:29:37 +02:00
Родитель 8ae1ef72fc
Коммит bacf5485e8
1 изменённых файлов: 1 добавлений и 2 удалений
Показать статистику Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы

3
train.py
Просмотреть файл

@ -264,7 +264,7 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
# ('linear_loss', linear_loss.item()), # ('linear_loss', linear_loss.item()),
# ('mel_loss', mel_loss.item()), # ('mel_loss', mel_loss.item()),
# ('stop_loss', stop_loss.item())]) # ('stop_loss', stop_loss.item())])
if current_step % c.print_step == 0: if num_iter % c.print_step == 0:
print(" | | > TotalLoss: {:.5f} LinearLoss: {:.5f} MelLoss:{:.5f} "\ print(" | | > TotalLoss: {:.5f} LinearLoss: {:.5f} MelLoss:{:.5f} "\
"StopLoss: {:.5f} ".format(loss.item(), "StopLoss: {:.5f} ".format(loss.item(),
linear_loss.item(), linear_loss.item(),
@ -319,7 +319,6 @@ def evaluate(model, criterion, criterion_st, data_loader, current_step):
def main(args): def main(args):
# Setup the dataset
# Setup the dataset # 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'),