stransformers

Environment setup

Setup pyenv using this article

# setup a Python3 virtual environment
pyenv global 3.8.6
mkvirtualenv stransformers
# OR
virtualenv stransformers -p `which python3`

Simple Transformers

A wrapper on Hugging Face to simplify model training steps

T5 Small Example

Used this blog post for training a new task with T5 small, referring to this git repo

Install the following packages, corresponding to the conda list/env here

# ------------------------------------------------------------------------------
pip install pandas tqdm
pip install torch torchvision
pip install simpletransformers
pip install jupyter

Adapted from Thilina Rajapakse's code to create a custom Jupyter notebook for our passive to active conversion task:p2a_T5_small.ipynb