/P-tuning-v2-MRC-NER

Primary LanguagePythonMIT LicenseMIT

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