fix training code bug

2021-10-13 00:05:06 +03:00 · 2020-11-11 13:47:10 -05:00
parent ad68251221
commit 2df8833965
1 changed files with 1 additions and 1 deletions
--- a/textattack/commands/train_model/run_training.py
+++ b/textattack/commands/train_model/run_training.py
@@ -250,7 +250,7 @@ def _generate_adversarial_examples(model, attack_class, dataset):

    :return: list(AttackResult) of adversarial examples.
    """
-    attack = attack_class(model)
+    attack = attack_class.build(model)

    try:
        # Fix TensorFlow GPU memory growth