From 1b7b06c75e88409c6afffea1cd878d150e98ba0c Mon Sep 17 00:00:00 2001 From: Raul Puri Date: Thu, 3 May 2018 18:29:37 -0700 Subject: [PATCH] model.py 0.4 update --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 1f9e7d1..404822e 100644 --- a/model.py +++ b/model.py @@ -471,7 +471,7 @@ class Tacotron2(nn.Module): output_lengths = batch text_padded = to_gpu(text_padded).long() input_lengths = to_gpu(input_lengths).long() - max_len = torch.max(input_lengths.data) + max_len = torch.max(input_lengths.data).item() mel_padded = to_gpu(mel_padded).float() gate_padded = to_gpu(gate_padded).float() output_lengths = to_gpu(output_lengths).long()