Browse Source

model.py: removing .item

master
Rafael Valle 6 years ago
parent
commit
84c83f96c9
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      model.py

+ 1
- 1
model.py View File

@ -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()

Loading…
Cancel
Save