- Pytorch please select the correct CUDA version matched your GPU.
- other requirements
pip install -r requirements.txt
If some packages are still missing, please use pip install <package_name>
python baseline.py
If you would like to change hyperparameters or models, please edit the file directly (too lazy to write CLI)
python train.py
python evaluate.py
python predict.py
Please check prompting folder