зеркало из https://github.com/mozilla/TTS.git
remove Variable from losses.py
This commit is contained in:
Родитель
0a2e9fef9e
Коммит
a856f76791
|
@ -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)
|
||||
|
|
Загрузка…
Ссылка в новой задаче