quaterion-models
quaterion-models copied to clipboard
The collection of bulding blocks building fine-tunable metric learning models
updates: - [github.com/pre-commit/pre-commit-hooks: v4.4.0 → v6.0.0](https://github.com/pre-commit/pre-commit-hooks/compare/v4.4.0...v6.0.0) - https://github.com/psf/black → https://github.com/psf/black-pre-commit-mirror - [github.com/psf/black-pre-commit-mirror: 22.12.0 → 25.12.0](https://github.com/psf/black-pre-commit-mirror/compare/22.12.0...25.12.0) - [github.com/PyCQA/isort: 5.11.4 → 7.0.0](https://github.com/PyCQA/isort/compare/5.11.4...7.0.0)
This PR fix a "list index out of range" issue when inferring from a Quaterion model with multiple encoders. This bug wasn't visible in the example provided as the number...
We need to test more cases in `quaterion-models` --the current coverage is too weak to catch regressions.
``` from quaterion_models import SimilarityModel from torch.cuda.amp import autocast from torch.utils.data import DataLoader, Dataset from torchvision import models as torch_models, transforms from tqdm import tqdm from PIL import Image class...
I have a project that I would like to upgrade to Python 3.11 to solve a compatibility issue with another dependency; however, it appears that quaterion-models only supports Python versions...