mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
4 lines
245 B
Bash
Executable File
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 |