From 84c83f96c9fe3e9de6a68d2a84032ef748ffdedb Mon Sep 17 00:00:00 2001 From: Rafael Valle Date: Thu, 3 May 2018 19:42:01 -0700 Subject: [PATCH] model.py: removing .item --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 404822e..1f9e7d1 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).item() + max_len = torch.max(input_lengths.data) mel_padded = to_gpu(mel_padded).float() gate_padded = to_gpu(gate_padded).float() output_lengths = to_gpu(output_lengths).long()