wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove Variable from losses.py

Browse Source
This commit is contained in:
Eren Golge
2018-05-10 16:27:55 -07:00
parent 0a2e9fef9e
commit a856f76791
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers/losses.py
-2
View File
@@ -1,6 +1,5 @@
import torch
from torch.nn import functional
from torch.autograd import Variable
from torch import nn
@@ -11,7 +10,6 @@ def _sequence_mask(sequence_length, max_len=None):
batch_size = sequence_length.size(0)
seq_range = torch.arange(0, max_len).long()
seq_range_expand = seq_range.unsqueeze(0).expand(batch_size, max_len)
seq_range_expand = Variable(seq_range_expand)
if sequence_length.is_cuda:
seq_range_expand = seq_range_expand.cuda()
seq_length_expand = (sequence_length.unsqueeze(1)