COMP 70016 CW
Python requirement
pip install transformers sklearn pandas
Gudience
To run training and evaluation just do:
cd NLP_CW
python NLP2077.py
Result
The results will be printed and model saved in NLP_CW/models
IMPORTANT
- We trained our model on an nvidia v100s GPU with 32GB of memory. Please change the batchsize on line 27 or use CPU if you are out of memory
Contact
- Yitao Qiu yq1321@ic.ac.uk
- Jonas Brami jb2021@ic.ac.uk
- Zifeng Tang zt520@ic.ac.uk