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

add batch size to config & log stderr output from tests

Browse Source
This commit is contained in:
Jack Morris
2020-05-08 15:53:57 -04:00
parent 53a5c3a7a1
commit f847e4c472
6 changed files with 38 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/goal_functions/goal_function.py
View File

@@ -18,7 +18,7 @@ class GoalFunction:
self.use_cache = use_cache
self.num_queries = 0
if self.use_cache:
self._call_model_cache = lru.LRU(2**18)
self._call_model_cache = lru.LRU(utils.config('MODEL_CACHE_SIZE'))
else:
self._call_model_cache = None