Skip to content

BarlowTwins Benchmark on ImageNet

Compare
Choose a tag to compare
@MalteEbner MalteEbner released this 29 Aug 10:54
· 251 commits to master since this release
6092e09

Changes

  • Added a new benchmark of BarlowTwins on ImageNet.
  • Optimized performance of the BarlowTwinsLoss computation, making it much faster
  • Fixed a bug in the CosineWarmupScheduler. Thanks to @anishacharya for pointing out the problem.
  • Cleaned the setup.py

Models