TextBrewer是一个基于PyTorch的、为实现NLP中的知识蒸馏任务而设计的工具包, 融合并改进了NLP和CV中的多种知识蒸馏技术,提供便捷快速的知识蒸馏框架, 用于以较低的性能损失压缩神经网络模型的大小,提升模型的推理速度,减少内存占用。
可以通过ACL Anthology或arXiv pre-print查看我们的论文。
TextBrewer完整文档
新闻
Dec 17, 2021
- 哈工大讯飞联合实验室全新推出模型裁剪工具包TextPruner。查看:GitHub - airaria/TextPruner: A PyTorch-based model pruning toolkit for pre-trained language models
Oct 24, 2021
- 哈工大讯飞联合