zhenwang9102 / soft-prompt-tuning

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The Power of Scale for Parameter-Efficient Prompt Tuning

Implementation of soft embeddings from https://arxiv.org/abs/2104.08691v1 using Pytorch and Huggingface transformers

Citation

@misc{lester2021power,
      title={The Power of Scale for Parameter-Efficient Prompt Tuning}, 
      author={Brian Lester and Rami Al-Rfou and Noah Constant},
      year={2021},
      eprint={2104.08691},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}

About

License:MIT License


Languages

Language:Jupyter Notebook 50.5%Language:Python 49.5%