qdrant / quaterion

Blazing fast framework for fine-tuning similarity learning models

Why this job: works in the areas we cover, but does not say which job it does

What it does

Warp-speed fast. With the built-in caching mechanism, Quaterion enables you to train thousands of epochs with huge batch sizes even on *laptop GPU*.
Small data compatible. Pre-trained models with specially designed head layers allow you to benefit even from a dataset you can label *in one day*.
Customizable. Quaterion allows you to re-define any part of the framework, making it flexible even for large-scale and sophisticated training pipelines.
Scalable. Quaterion is built on top of PyTorch Lightning and inherits all its scalability, cost-efficiency, and reliability perks.

pip install quaterion

Repos similar to this one
Repo	Stars	Last commit	Licence	Downloads for
KevinMusgrave / pytorch-metric-learning	6,343	4 days ago	MIT	No downloads
scikit-learn-contrib / metric-learn	1,430	7 months ago	MIT	No downloads
gorse-io / gorse	9,921	13 days ago	Apache-2.0	Mac, Windows, Linux

More: languages, topics, project record and preview image