Skip to content

v0.9.89
Compare
Choose a tag to compare
@KevinMusgrave KevinMusgrave released this 25 Jul 14:40
· 956 commits to master since this release
v0.9.89
a57ed4e

CrossBatchMemory

  • Fixed bug where CrossBatchMemory would use self-comparisons as positive pairs. This was uniquely a CrossBatchMemory problem because of the nature of adding each current batch to the queue.
  • Fixed bug where DistanceWeightedMiner would not work with CrossBatchMemory due to missing ref_label
  • Changed 3rd keyword argument of forward() from input_indices_tuple to indices_tuple to be consistent with all other losses.

AccuracyCalculator

  • Fixed bug in AccuracyCalculator where it would return NaN if the reference set contained none of query set labels. Now it will log a warning and return 0.

BaseTester

  • Fixed bug where "compared_to_training_set" mode of BaseTester fails due to list(None) bug.

InferenceModel

  • New get_nearest_neighbors function will return nearest neighbors of a query. By @btseytlin

Loss and miner utils

  • Switched to fill_diagonal_ in the get_all_pairs_indices and get_all_triplets_indices code, instead of creating torch.eye.
Assets 2
Loading