From 56bbd95d9adb50d4786cf4022086a214d9d63862 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Wed, 25 Apr 2018 08:42:21 -0700 Subject: [PATCH] bug fix Q ~ ' --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 0c069607..f63783fb 100644 --- a/train.py +++ b/train.py @@ -264,7 +264,7 @@ def evaluate(model, criterion, data_loader, current_step): avg_mel_loss += mel_loss.item() # Diagnostic visualizations - idx = np.random.randint(mel_input.shape[0]) + idx = np.random.randint(mel_spec.shape[0]) const_spec = linear_output[idx].data.cpu().numpy() gt_spec = linear_spec[idx].data.cpu().numpy() align_img = alignments[idx].data.cpu().numpy()