OpenPrompt

收录理由

OpenPrompt 是一个基于 PyTorch 和 Hugging Face transformers 的提示学习框架。它把模板（围绕输入构造的文本）与言语器（标签词到模型词汇表的映射）分开设计，因此你可以在不重写实验代码的前提下，快速测试不同的提示思路。切换模型时，比如从 BERT 换成 RoBERTa 或 T5，只需改动模型参数，而无需调整整个流程。库内自带常用 NLP 任务的数据处理器和训练器，当你想要复现提示学习论文中的结果，而不是从零搭建脚手架时，能节省大量时间。它还支持与 UltraChat 配合进行监督指令微调，并获得了 ACL 2022 最佳演示论文奖。对于需要在多个模型间比较提示策略的团队来说，这是一个非常实用的基础工具。