Skip to content

Commit 567079b

Browse files
committed
Create train_lstm_rotten_tomatoes_sentiment_classification.sh
1 parent 7200d7f commit 567079b

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
#!/bin/bash
2+
# Trains `bert-base-cased` on the STS-B task for 3 epochs. This is a basic
3+
# demonstration of our training script and `datasets` integration.
4+
textattack train --model-name-or-path lstm --dataset rotten_tomatoes --epochs 50 --learning-rate 1e-5

0 commit comments

Comments
 (0)