Skip to content

Latest commit

 

History

History
16 lines (13 loc) · 392 Bytes

README.md

File metadata and controls

16 lines (13 loc) · 392 Bytes

P-tuning-MRC-NER

Env

python==3.9.15 pytorch==1.10.2 pytorch-lightning==0.9.0 transformers==4.15.0

Install depedencies

  • start with a clean python3 virtual environment with conda
  • install pytorch with conda install pytorch==1.10.2 pytorch-cuda=11.6 -c pytorch -c nvidia
  • install rest of the packages:
pip install pytorch-lightning==0.9.0 transformers==4.15.0 tokenizers