Remove unused variables in examples.

2019-12-23 22:23:44 +01:00
parent 072750f4dc
commit 81422c4e6d
4 changed files with 2 additions and 16 deletions
--- a/examples/summarization/modeling_bertabs.py
+++ b/examples/summarization/modeling_bertabs.py
@@ -446,8 +446,6 @@ class MultiHeadedAttention(nn.Module):
        batch_size = key.size(0)
        dim_per_head = self.dim_per_head
        head_count = self.head_count
-        key_len = key.size(1)
-        query_len = query.size(1)

        def shape(x):
            """  projection """
@@ -504,9 +502,6 @@ class MultiHeadedAttention(nn.Module):

        query = shape(query)

-        key_len = key.size(2)
-        query_len = query.size(2)
-
        # 2) Calculate and scale scores.
        query = query / math.sqrt(dim_per_head)
        scores = torch.matmul(query, key.transpose(2, 3))