一个有价值的模型蒸馏工具-TextBrewer

TextBrewer is a PyTorch-based model distillation toolkit for natural language processing. It includes various distillation techniques from both NLP and CV field and provides an easy-to-use distillation framework, which allows users to quickly experiment with the state-of-the-art distillation methods to compress the model with a relatively small sacrifice in the performance, increasing the inference speed and reducing the memory usage.

地址:https://github.com/airaria/TextBrewer

如若有时间,可以研究一番,将其应用到工程中。

 

posted @ 2021-06-29 15:11  今夜无风  阅读(242)  评论(0编辑  收藏  举报