From a856f76791787ed67df7d0e6f81cf809925adc06 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Thu, 10 May 2018 16:27:32 -0700 Subject: [PATCH] remove Variable from losses.py --- layers/losses.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/layers/losses.py b/layers/losses.py index 3fbd3141..2135ee89 100644 --- a/layers/losses.py +++ b/layers/losses.py @@ -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)