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

fix beam search, _->-

Browse Source
This commit is contained in:
uvafan
2020-05-19 14:09:40 -04:00
parent f1d0017a53
commit 30cd452ed1
3 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/shared/scripts/run_attack_args_helper.py
View File

@@ -211,7 +211,7 @@ def get_args():
attack_group = parser.add_mutually_exclusive_group(required=False)
search_choices = ', '.join(SEARCH_CLASS_NAMES.keys())
attack_group.add_argument('--search', '-s', '--search_method', type=str,
attack_group.add_argument('--search', '-s', '--search-method', type=str,
required=False, default='greedy-word-wir',
help=f'The search method to use. choices: {search_choices}')