Entailment needed [CLS] token. Now all tests pass!

2021-10-13 00:05:06 +03:00 · 2020-04-29 17:59:48 -04:00
parent f47734dff7
commit 57b11e74a0
4 changed files with 3 additions and 4 deletions
--- a/textattack/goal_functions/goal_function.py
+++ b/textattack/goal_functions/goal_function.py
@@ -43,7 +43,6 @@ class GoalFunction:
        display purposes, and a score.
        """
        model_outputs = self._call_model(tokenized_text_list)
-        import pdb; pdb.set_trace()
        results = []
        for tokenized_text, raw_output in zip(tokenized_text_list, model_outputs):
            succeeded = self._is_goal_complete(raw_output, ground_truth_output)