Requirements: Python 3.8, PyTorch 1.8.2, CUDA 11.3
pip install -r requirements.txt
cd src
python main.py [ rush | pretrain | train | train_full | train_ssl | test ]Requirements: Python 3.8, PyTorch 1.8.2, CUDA 11.3
pip install -r requirements.txt
cd src
python main.py [ rush | pretrain | train | train_full | train_ssl | test ]