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): def forward(self, query, inputs, processed_inputs, mask):
if self.location_attention: if self.location_attention:
attention, processed_query = self.get_location_attention( attention, _ = self.get_location_attention(
query, processed_inputs) query, processed_inputs)
else: else:
attention, processed_query = self.get_attention( attention, _ = self.get_attention(
query, processed_inputs) query, processed_inputs)
# apply masking # apply masking
if mask is not None: if mask is not None: