项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化，观测越久曲线越完整
数据来源：GitHub API 逐日快照 · 收录于 2026-09-16 · 当前星标 ★ 665
使用教程
环境要求
- Python 3.9+
- PyTorch
- 成对标注数据
安装与启动步骤
-
1安装
相似度学习模型的微调框架，pip 装核心。
pip install quaterion -
2装模型依赖
预训练模型单独一个包。
pip install quaterion-models -
3准备三元组数据
相似学习依赖正负样本对，数据质量决定效果。
-
4训练并评估
用验证集看嵌入的区分度，别只看 loss。
如何确认成功
训练后相似样本的嵌入距离明显更近
注意事项
- 负样本采样策略影响很大，随机采样往往不够。
- 评估要看检索指标而非分类准确率。
核心亮点
- 内置多种对比/度量学习损失与采样器，减少手写训练循环的样板代码
- 针对训练速度做了工程优化，官方基准显示比同类框架更快
- 基于 PyTorch，编码器、数据加载器和评估器均可自定义，灵活度较高
不足之处
- 社区规模较小，生态和第三方教程相对有限
- 文档和示例覆盖度一般，复杂场景需阅读源码
适用场景
- 电商商品去重与相似商品召回模型微调
- 文本语义检索或问答匹配的嵌入模型训练
- 图像/多模态场景下的近邻搜索向量表示学习
替代项目
sentence-transformers、pytorch-metric-learning
下一篇：physicsnemo
同类项目推荐
finetune-SAM
开源
把通用分割模型SAM微调到你的医学图像上，提升精度
This is an official repo for fine-tuning SAM to customized medical images.
AMD-LLM
开源
在 AMD GPU 上快速跑通 1.35 亿参数语言模型训练
Training code and resources for AMD-135M language models on AMD GPUs.
Prism
开源
动态稀疏注意力，让2K音视频生成模型原生训练更省算力
We propose Prism, a dynamic sparse attention framework for natively training joint v···
rlix
开源
让 RL 实验排队更聪明，GPU 等待时间更短
Run more RL experiments. Wait less for GPUs.
aimet
开源
不重训模型，把已训练网络压到 4bit 还能保住精度
AIMET is a library that provides advanced quantization and compression techniques fo···
JetSpec
开源
让大模型推理更快：并行草稿树突破推测解码速度天花板
JetSpec++/JetSpec: Causal Parallel (Tree) Drafting with Self Refinement
finetrainers
开源
用有限显存微调视频扩散模型，多卡也能跑
Scalable and memory-optimized training of diffusion models
AutoGPTQ
开源
一键量化大模型，显存减半推理加速
An easy-to-use LLMs quantization package with user-friendly apis, based on GPTQ algo···
diffusion-DDIM-pytorch
开源
快速上手 DDIM，用更少步数生成高质量图像
This is a pytorch implementation of Denoising Diffusion Implicit Models