Skip to content

Releases: KevinMusgrave/pytorch-metric-learning

v2.6.0

24 Jul 12:39
28d5acc
Compare
Choose a tag to compare

Improvement + small breaking change to DistributedLossWrapper

  • Changed the emb argument of DistributedLossWrapper.forward to embeddings to be consistent with the rest of the library.
  • Added a warning and early-return when DistributedLossWrapper is being used in a non-distributed setting.
  • Thank you @elisim!

v2.5.0

01 Apr 08:16
adfb78c
Compare
Choose a tag to compare

v2.4.1

16 Dec 19:24
3a14f82
Compare
Choose a tag to compare

This is identical to v2.4.0, but includes the LICENSE file which was missing from v2.4.0.

v2.4.0

16 Dec 04:55
e540cd0
Compare
Choose a tag to compare

Features

Bug fixes

  • Fixed issue where PNPLoss would return NaN when a batch sample had no corresponding positive. See PR #660. Thanks @Puzer and @interestingzhuo!

Tests

  • Fixed the test for HistogramLoss to work with PyTorch 2.1. Thanks @GaetanLepage!

v2.3.0

25 Jul 15:01
ac60700
Compare
Choose a tag to compare

Features

v2.2.0

18 Jun 18:51
Compare
Choose a tag to compare

Features

  • Added ManifoldLoss. See pull request #635. Thanks @domenicoMuscill0!
  • Added P2SGradLoss. See pull request #635. Thanks @domenicoMuscill0!
  • Added the symmetric flag to SelfSupervisedLoss. If True, then the embeddings in both embeddings and ref_emb are used as anchors. If False, then only the embeddings in embeddings are used as anchors. The previous behavior was equivalent to symmetric=False. Now the default is symmetric=True, because this is usually what is done in self supervised papers (e.g. SimCLR).

v2.1.2

27 May 21:34
Compare
Choose a tag to compare

Bug Fixes

  • Fixed bug where set_stats was not being called in TripletMarginMiner (#628)
  • Made HierarchicalSampler extend torch.utils.data.Sampler instead of torch.utils.data.BatchSampler (#613)
  • Made samplers documentation clearer (#615). Thanks @rheum !

v2.1.1

03 May 11:58
c57ebdd
Compare
Choose a tag to compare

Bug Fixes

  • Fixes bug where BaseDistance.initial_avg_query_norm was not actually being set (#620)

v2.1.0

05 Apr 23:21
03d5192
Compare
Choose a tag to compare

Features

New loss function: PNPLoss. Thanks @interestingzhuo!

v2.0.1

21 Feb 17:34
691a635
Compare
Choose a tag to compare

Bug Fixes