unused var

This commit is contained in:
Thomas Werkmeister 2019-07-23 19:33:56 +02:00
parent 98edb7a4f8
commit 82db35530f
1 changed files with 2 additions and 2 deletions

View File

@ -234,10 +234,10 @@ class Attention(nn.Module):
def forward(self, query, inputs, processed_inputs, mask):
if self.location_attention:
attention, processed_query = self.get_location_attention(
attention, _ = self.get_location_attention(
query, processed_inputs)
else:
attention, processed_query = self.get_attention(
attention, _ = self.get_attention(
query, processed_inputs)
# apply masking
if mask is not None: