Skip to content

Releases: KevinMusgrave/pytorch-metric-learning

v1.5.1

16 Jul 19:39
Compare
Choose a tag to compare

Bug fixes

Bumped the record-keeper version to fix issue #497

v1.5.0

29 Jun 21:58
Compare
Choose a tag to compare

Features

For some loss functions, labels are now optional if indices_tuple is provided:

loss = loss_func(embeddings, indices_tuple=pairs)

The losses for which you can do this are:

  • CircleLoss
  • ContrastiveLoss
  • IntraPairVarianceLoss
  • GeneralizedLiftedStructureLoss
  • LiftedStructureLoss
  • MarginLoss
  • MultiSimilarityLoss
  • NTXentLoss
  • SignalToNoiseRatioContrastiveLoss
  • SupConLoss
  • TripletMarginLoss
  • TupletMarginLoss

This issue has come up several times:

#412
#490
#482
#473
#179
#263

v1.4.0

09 Jun 18:03
6f65823
Compare
Choose a tag to compare

New features

v1.3.2

29 May 15:26
Compare
Choose a tag to compare

Bug fixes

  • Fixed a bug in BatchEasyHardMiner where get_max_per_row was not always returning correct values, resulting in invalid pairs and triplets. #476

v1.3.1

27 May 10:51
Compare
Choose a tag to compare

Bug fixes

  • Fixed ThresholdReducer being incompatible with older versions of PyTorch (#465)
  • Fixed VICRegLoss being incompatible with older versions of PyTorch, and missing a division by 2 (#467 and #470 by @cwkeam)

Other

  • Made CustomKNN more memory efficient by removing torch.cat call.

v1.3.0

30 Mar 08:31
Compare
Choose a tag to compare

Features

  • Added a batch_size parameter to CustomKNN. This computes k-nn per batch of query embeddings (using BatchedDistance), which requires less memory than computing the entire distance matrix at once.

Bug Fixes

v1.2.1

17 Mar 04:13
a142af3
Compare
Choose a tag to compare

Bug Fixes

v1.2.0

01 Mar 02:00
8b939bb
Compare
Choose a tag to compare

New Loss Function: SubCenterArcFace

Thanks @chingisooinar!

v1.1.2

16 Feb 08:15
0271f52
Compare
Choose a tag to compare

Bug fixes

v1.1.1

12 Feb 15:04
df53e74
Compare
Choose a tag to compare

Bug fixes