Code

Igor lig 4447 w mse benchmark#1474

Code

Open

IgorSusmelj wants to merge 13 commits intomasterlightly-ai/lightly:masterfrom igor-lig-4447-w-mse-benchmarklightly-ai/lightly:igor-lig-4447-w-mse-benchmarkCopy head branch name to clipboard

+169-3

Commits

Commits on Jan 11, 2024

• Add projection head

  IgorSusmelj

  committed
  c003c32

  Copy full SHA for c003c32

• Add ImageNet benchmark code for wmse

  IgorSusmelj

  committed
  d84b0e5

  Copy full SHA for d84b0e5

• Add wmse model to main script

  IgorSusmelj

  committed
  753cd5c

  Copy full SHA for 753cd5c

• Fix head

  IgorSusmelj

  committed
  6d53a0a

  Copy full SHA for 6d53a0a

• Add better error messages. Support mixed precision and distributed.

  IgorSusmelj

  committed
  dc99515

  Copy full SHA for dc99515

• Update code

  IgorSusmelj

  committed
  8b19156

  Copy full SHA for 8b19156

Commits on Jan 15, 2024

• Update comment to make clear why we have to use fp32

  IgorSusmelj

  committed
  827c8af

  Copy full SHA for 827c8af

• Use parmeters from paper

  IgorSusmelj

  committed
  e8b9d7e

  Copy full SHA for e8b9d7e

• Remove distributed gathering

  IgorSusmelj

  committed
  f5a913c

  Copy full SHA for f5a913c

Commits on Jan 19, 2024

• Add W-MSE results

  IgorSusmelj

  committed
  d4457dc

  Copy full SHA for d4457dc

• Add W-MSE results

  IgorSusmelj

  committed
  a798dcb

  Copy full SHA for a798dcb

Commits on Jan 21, 2024

• Add distributed gather again

  IgorSusmelj

  committed
  0fbde50

  Copy full SHA for 0fbde50

• Update code to match ImageNet experiments

  IgorSusmelj

  committed
  7ad8205

  Copy full SHA for 7ad8205