欢迎来到 Quaterion 文档!¶
Quaterion 是一个用于微调相似性学习模型的框架。该框架解决了为语义搜索、推荐、异常检测、极端分类、匹配引擎等训练模型的“最后一公里”问题。
它旨在将预训练模型的性能与自定义任务的专业化相结合,同时避免缓慢且昂贵的训练。
特性¶
🌀 极速:借助内置缓存机制,Quaterion 使您即使在笔记本电脑 GPU 上也能使用巨大的批处理大小训练数千个 epoch。
🐈 小数据兼容:带有特殊设计的头部层的预训练模型让您即使从可以在一天内标注的数据集中也能受益。
🏗️ 可定制:Quaterion 允许您重新定义框架的任何部分,使其即使对于大规模和复杂的训练管道也具有灵活性。
🌌 可扩展:Quaterion 构建于 PyTorch Lightning 之上,并继承了其所有的可扩展性、成本效率和可靠性优势。
安装¶
TL;DR
用于训练
pip install quaterion
用于推理服务
pip install quaterion-models
Quaterion 框架由两个包组成 - quaterion 和 quaterion-models。
由于并非总是可能或方便以 ONNX 格式表示模型(而且,它受支持),Quaterion 在一个单独的包中保留了对模型推理可能需要的一组非常精简的模型类。
这样可以避免在推理基础设施中安装沉重的训练依赖项:pip install quaterion-models
同时,一旦您需要用于模型训练和调试的完整工具集,它可以在一个包中获得:pip install quaterion
下一步¶
Quaterion API