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 multilabel load for pre-trained lstm/cnn; add benchmarked results for MR

Browse Source
This commit is contained in:
Jack Morris
2020-07-07 21:16:39 -04:00
parent c0a3a734d8
commit 7e3b90e979
7 changed files with 203 additions and 191 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -322,7 +322,11 @@ def parse_model_from_args(args):
model_from_args,
)
model = model_from_args(argparse.Namespace(**model_train_args), num_labels)
model = model_from_args(
argparse.Namespace(**model_train_args),
num_labels,
model_path=args.model,
)
else:
raise ValueError(f"Error: unsupported TextAttack model {args.model}")
return model