🐛 fix cls => sep HIT-SCIR#538

jrjdr · Sep 30, 2021 · 534de4e · 534de4e
1 parent f3d4a25
commit 534de4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ltp/transformer_rel_linear.py b/ltp/transformer_rel_linear.py
@@ -46,7 +46,7 @@ def forward(self, input, attention_mask=None, word_index=None, word_attention_ma
         if not is_processed:
             if not self.use_cls:
                 input = input[:, 1:, :]
-            if not self.use_cls:
+            if not self.use_sep:
                 input = input[:, :-1, :]
 
             if word_attention_mask is None: