1
0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Files
textattack-nlp-transformer/examples/train/train_lstm_imdb_sentiment_classification.sh
2021-08-02 16:42:49 -04:00

4 lines
245 B
Bash
Executable File

#!/bin/bash
# Trains `bert-base-cased` on the STS-B task for 3 epochs. This is a basic
# demonstration of our training script and `datasets` integration.
textattack train --model-name-or-path lstm --dataset imdb --epochs 50 --learning-rate 1e-5