3 ms·
Not a full featured framework, but pytorch-metric-learning has data loaders, lossess, etc. to facilitate similarity learning: https://github.com/KevinMusgrave/p
by flaviojuvenal 4y ago
Not a full featured framework, but pytorch-metric-learning has data loaders, lossess, etc. to facilitate similarity learning: https://github.com/KevinMusgrave/pytorch-metric-learning https://github.com/KevinMusgrave/pytorch-metric-learning
Disclaimer: I've made some contributions to it.
- dimitry12 4y agoSeconding this. PML is high quality, active, and well documented.
- generall 4y agoPML is a great collection of implementations, but not the best framework. Also you can use PML with Quaterion: https://github.com/qdrant/quaterion/blob/master/examples/train_cifar100_with_pytorch_metric_learning.py https://github.com/qdrant/quaterion/blob/master/examples/tra...