alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

PR fixes

Browse Source
This commit is contained in:
Jack Morris
2020-05-02 20:55:51 -04:00
parent 5d6c43581c
commit cc665f9aaa
9 changed files with 14 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
textattack/goal_functions/goal_function.py
View File

@@ -139,8 +139,8 @@ class GoalFunction:
outputs = self._call_model_uncached(uncached_list)
for text, output in zip(uncached_list, outputs):
self._call_model_cache[text] = output
final_scores = [self._call_model_cache[text] for text in tokenized_text_list]
return final_scores
all_outputs = [self._call_model_cache[text] for text in tokenized_text_list]
return all_outputs
def extra_repr_keys(self):
return []