Skip to content

BYOLTransform

Compare
Choose a tag to compare
@guarin guarin released this 25 Jul 09:38
· 225 commits to master since this release
80b1583

Changes

  • Add BYOLTransform which replaces SimCLRTransform in BYOL benchmarks.
  • Log benchmark results only on rank0.
  • Fix bug in PMSNLoss where probabilities were not converted to log-space before the loss calculation. Thanks to @Cloudy1225 for reporting this!
  • Version check now runs in background and no longer requires SIGALRM.
  • Add support for scheduling Lightly Worker runs with the new selection strategy strength option.

Models