[66af30]: / scripts / train_text_sentiment_classifier.sh

Download this file

8 lines (8 with data), 219 Bytes

1
2
3
4
5
6
7
8
python3 train_sentiment_textbased.py \
--dataset_name SST \
--model_name pretrain_Bart \
--num_epoch 20 \
-lr 0.0001 \
-b 32 \
-s ./checkpoints/text_sentiment_classifier \
-cuda cuda:0