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

support custom models; waiting on datasets integration

Browse Source
This commit is contained in:
Jack Morris
2020-06-05 17:35:52 -04:00
parent 20f3769963
commit b6dd920c08
16 changed files with 351 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/shared/validators.py
View File

@@ -59,7 +59,7 @@ def validate_model_goal_function_compatibility(goal_function_class, model_class)
# Otherwise, this is an unknown model–perhaps user-provided, or we forgot to
# update the corresponding dictionary. Warn user and return.
logger.warn(f'Unknown if model {model} compatible with goal function {goal_function}.')
logger.warn(f'Unknown if model of class {model_class} compatible with goal function {goal_function}.')
return True
def validate_model_gradient_word_swap_compatibility(model):