fix a bug

xyyhcl · Nov 17, 2020 · 7440a17 · 7440a17
1 parent 989de3f
commit 7440a17
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 40 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git a/bert_seq2seq/model/bert_model.py b/bert_seq2seq/model/bert_model.py
@@ -104,11 +104,9 @@ def __init__(self, config):
         self.dropout = nn.Dropout(config.hidden_dropout_prob)
 
     def forward(self, input_ids=None, token_type_ids=None, position_ids=None):
-        if input_ids is not None:
-            input_shape = input_ids.size()
-        else:
-            input_shape = inputs_embeds.size()[:-1]
-
+
+        input_shape = input_ids.size()
+
         seq_length = input_shape[1]
         device = input_ids.device
         if position_ids is None:
@@ -446,7 +444,6 @@ def forward(
         output_attentions=False
     ):  
 
-
         extended_attention_mask = (input_ids > 0).float()
         # 注意力矩阵mask: [batch_size, 1, 1, seq_length]
         extended_attention_mask = extended_attention_mask.unsqueeze(1).unsqueeze(2)

diff --git a/test/test.py b/test/test.py