mirror of https://github.com/coqui-ai/TTS.git
fix #638
This commit is contained in:
parent
9646624cd5
commit
83143fbe39
|
@ -200,7 +200,7 @@ def train(model, optimizer, criterion, scheduler, scaler, ap, global_step, epoch
|
||||||
train_data[rand_idx], (tuple, list)) else train_data[rand_idx][0]
|
train_data[rand_idx], (tuple, list)) else train_data[rand_idx][0]
|
||||||
wav = ap.load_wav(wav_path)
|
wav = ap.load_wav(wav_path)
|
||||||
ground_mel = ap.melspectrogram(wav)
|
ground_mel = ap.melspectrogram(wav)
|
||||||
sample_wav = model.generate(ground_mel,
|
sample_wav = model.inference(ground_mel,
|
||||||
c.batched,
|
c.batched,
|
||||||
c.target_samples,
|
c.target_samples,
|
||||||
c.overlap_samples,
|
c.overlap_samples,
|
||||||
|
@ -287,7 +287,7 @@ def evaluate(model, criterion, ap, global_step, epoch):
|
||||||
eval_data[rand_idx], (tuple, list)) else eval_data[rand_idx][0]
|
eval_data[rand_idx], (tuple, list)) else eval_data[rand_idx][0]
|
||||||
wav = ap.load_wav(wav_path)
|
wav = ap.load_wav(wav_path)
|
||||||
ground_mel = ap.melspectrogram(wav)
|
ground_mel = ap.melspectrogram(wav)
|
||||||
sample_wav = model.generate(ground_mel,
|
sample_wav = model.inference(ground_mel,
|
||||||
c.batched,
|
c.batched,
|
||||||
c.target_samples,
|
c.target_samples,
|
||||||
c.overlap_samples,
|
c.overlap_samples,
|
||||||
|
|
Loading…
Reference in New Issue