Here is an example of using this package.
- Install requirements
pip install -r /path/to/requirements.txt
- Train RoBERTa-large model
python train_bert.py --model-name roberta-large --dataset-path ./data/1preproc.tsv
- The plots of the training:
-
roberta-large, uncased (data/1preproc.tsv) dataset:
-
roberta-large, cased (data/final.tsv) dataset:
-
- Olha Dovbnia [email protected]
- Witold Sosnowski [email protected]
- Anna Wróblewska [email protected]