zysNLP / transformers2TensorRT

本项目介绍了使用Transformers模块微调文本分类模型,以及模型转pth格式,进而转为onnx格式和TensorRT的engine格式,最后通过TensorRT进行推理的完整流程
7 stars 1 forks source link