From 63eea4a36480dede6d0f492b7cdaa6bacc3a6612 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 6 Jun 2019 10:24:34 +0200 Subject: [PATCH] bug fix --- .compute | 2 +- utils/synthesis.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.compute b/.compute index 8fa63c04..54c96861 100644 --- a/.compute +++ b/.compute @@ -12,5 +12,5 @@ python3 setup.py develop # cp -R ${USER_DIR}/GermanData ../tmp/ # python3 distribute.py --config_path config_tacotron_de.json --data_path ../tmp/GermanData/karlsson/ cp -R ${USER_DIR}/Mozilla_22050 ../tmp/ -python3 distribute.py --config_path config_tacotron.json --data_path ../tmp/Mozilla_22050/ --restore_path /data/rw/home/4845.pth.tar +python3 distribute.py --config_path config_tacotron_gst.json --data_path ../tmp/Mozilla_22050/ while true; do sleep 1000000; done diff --git a/utils/synthesis.py b/utils/synthesis.py index 4f4386ff..913c84ad 100644 --- a/utils/synthesis.py +++ b/utils/synthesis.py @@ -46,7 +46,7 @@ def synthesis(model, text, CONFIG, use_cuda, ap, truncated=False, enable_eos_bos decoder_output = decoder_output[0].data.cpu().numpy() alignment = alignments[0].cpu().data.numpy() # plot results - if CONFIG.model == "Tacotron": + if CONFIG.model in ["Tacotron", "TacotronGST"]: wav = ap.inv_spectrogram(postnet_output.T) else: wav = ap.inv_mel_spectrogram(postnet_output.T)