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 MR models and SNLI

Browse Source
This commit is contained in:
Jack Morris
2020-06-28 19:22:57 -04:00
parent 3f15bb34f8
commit c0edd5e08b
2 changed files with 17 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
textattack/commands/attack/attack_args_helpers.py
View File

@@ -333,9 +333,10 @@ def parse_dataset_from_args(args):
_, args.dataset_from_nlp = HUGGINGFACE_DATASET_BY_MODEL[args.model]
elif args.model in TEXTATTACK_DATASET_BY_MODEL:
_, args.dataset_from_nlp = TEXTATTACK_DATASET_BY_MODEL[args.model]
# Automatically detect dataset for models trained with textattack.
if args.model and os.path.exists(args.model):
elif args.model and os.path.exists(args.model):
model_args_json_path = os.path.join(args.model, "train_args.json")
if not os.path.exists(model_args_json_path):
raise FileNotFoundError(