From 64f2b95c31493e7d2df0902083e224a98294c403 Mon Sep 17 00:00:00 2001 From: Eren Golge Date: Tue, 13 Aug 2019 12:14:34 +0200 Subject: [PATCH] update regarding torch 1.2 --- layers/common_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers/common_layers.py b/layers/common_layers.py index 2edf0dab..4269728f 100644 --- a/layers/common_layers.py +++ b/layers/common_layers.py @@ -240,7 +240,7 @@ class Attention(nn.Module): attention_hidden_state, processed_inputs) # apply masking if mask is not None: - attention.data.masked_fill_(1 - mask, self._mask_value) + attention.data.masked_fill_(torch.bitwise_not(mask), self._mask_value) # apply windowing - only in eval mode if not self.training and self.windowing: attention = self.apply_windowing(attention, inputs)