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

update test script; batch size -> config.json

Browse Source
This commit is contained in:
Jack Morris
2020-05-09 09:48:27 -04:00
parent 581d936fe9
commit 1fc809e662
3 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/goal_functions/goal_function.py
View File

@@ -75,7 +75,7 @@ class GoalFunction:
"""
raise NotImplementedError()
def _call_model_uncached(self, tokenized_text_list, batch_size=16):
def _call_model_uncached(self, tokenized_text_list, batch_size=utils.config('MODEL_BATCH_SIZE')):
""" Queries model and returns outputs for a list of TokenizedText
objects.
"""