Update test_train.py

QData · Jun 5, 2022 · ce16561 · ce16561
1 parent dce49a0
commit ce16561
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_command_line/test_train.py b/tests/test_command_line/test_train.py
@@ -5,7 +5,7 @@
 
 
 def test_train_tiny():
-    command = "textattack train --model-name-or-path lstm --attack deepwordbug --dataset glue^cola --model-max-length 32 --num-epochs 2 --num-clean-epochs 1 --num-train-adv-examples 200"
+    command = "textattack train --model distilbert-base-uncased --attack textfooler --dataset rotten_tomatoes --model-max-length 64  --num-epochs 2 --num-clean-epochs 1 --num-train-adv-examples 10"
 
     # Run command and validate outputs.
     result = run_command_and_get_result(command)