Skip to content

v1.1.0
Compare
Choose a tag to compare
@KevinMusgrave KevinMusgrave released this 28 Dec 04:39
· 451 commits to master since this release
v1.1.0
429a309
This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
GPG key ID: 4AEE18F83AFDEB23
Expired
Verified
Learn about vigilant mode.

New features

CentroidTripletLoss

Implementation of On the Unreasonable Effectiveness of Centroids in Image Retrieval

VICRegLoss

Implementation of VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learning

AccuracyCalculator

  • Added mean reciprocal rank as an accuracy metric. Available as "mean_reciprocal_rank".
  • Added return_per_class argument for AccuracyCalculator. This is like avg_of_avgs but returns the accuracy per class, instead of averaging them for you.

Related issues

#369
#372
#374
#394

Contributors

Thanks to @cwkeam and @mlw214!

Contributors

mlw214 and cwkeam
Assets 2
Loading