unused var

2026-09-09 11:16:00 +08:00 · 2019-07-23 19:33:56 +02:00
parent 98edb7a4f8
commit 82db35530f
1 changed files with 2 additions and 2 deletions
@@ -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: