- Tested with Python 3.7.5
Step1. Install PyTorch: https://pytorch.org/get-started/locally/
# E.g., Linux, pip, NVidia GPU CUDA 11.7
pip3 install torch torchvision torchaudio
Step2. Install all requirements
pip install -r requirements.txt
python train_and_eval_T5-large.py
python evaluate.py