Code

Revision of BYOL module and tests#874

Code

Merged

otaj merged 24 commits intoLightning-Universe:masterLightning-Universe/lightning-bolts:masterfrom matsumotosan:byol_modulematsumotosan/lightning-bolts:byol_moduleCopy head branch name to clipboardSep 28, 2022

+166-136

Commits

Commits on Aug 25, 2022

• Remove review tag. Update docstring.

  matsumotosan

  committed
  70b2829

  Copy full SHA for 70b2829

• Test with SimCLR transforms instead of CPC.

  matsumotosan

  committed
  de54fb8

  Copy full SHA for de54fb8

• Add method to calculate loss for given online and target view. Update class docstrings.

  matsumotosan

  committed
  e097b0c

  Copy full SHA for e097b0c

• Add initial value of tau as an argument

  matsumotosan

  committed
  213ba1f

  Copy full SHA for 213ba1f

• Add encode method to calculate representation without calculating projection

  matsumotosan

  committed
  815def1

  Copy full SHA for 815def1

Commits on Aug 26, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  d185785

  Copy full SHA for d185785

• Rename losses to clarify direction

  matsumotosan

  committed
  198fa95

  Copy full SHA for 198fa95

Commits on Aug 28, 2022

• Rename projector_hidden_size to projector_hidden_dim

  matsumotosan

  committed
  c26a98a

  Copy full SHA for c26a98a

Commits on Sep 12, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  193223e

  Copy full SHA for 193223e

Commits on Sep 15, 2022

• Merge branch 'master' into byol_module

  otaj

  authored
  54f2630

  Copy full SHA for 54f2630

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  ab638ef

  Copy full SHA for ab638ef

Commits on Sep 16, 2022

• Merge branch 'master' into byol_module

  otaj

  authored
  aac6376

  Copy full SHA for aac6376

Commits on Sep 19, 2022

• Merge branch 'master' into byol_module

  Borda

  authored
  1e3cc4f

  Copy full SHA for 1e3cc4f

Commits on Sep 20, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  1c013b8

  Copy full SHA for 1c013b8

Commits on Sep 21, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  7b0682e

  Copy full SHA for 7b0682e

Commits on Sep 22, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  3a443f9

  Copy full SHA for 3a443f9

Commits on Sep 23, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  22ed1e5

  Copy full SHA for 22ed1e5

Commits on Sep 24, 2022

• Merge branch 'master' into byol_module

  matsumotosan

  authored
  bb1ada9

  Copy full SHA for bb1ada9

• Move logging to shared_step. Remove unused arguments from init.

  matsumotosan

  committed
  5414a06

  Copy full SHA for 5414a06

• Use CIFAR10 datamodule add dataset specific args. Fix BYOL module docs.

  matsumotosan

  committed
  2abd592

  Copy full SHA for 2abd592

• Add typing to parser argument

  matsumotosan

  committed
  1443878

  Copy full SHA for 1443878

Commits on Sep 27, 2022

• fix tests

  

  otaj

  committed
  cd10474

  Copy full SHA for cd10474

• Remove byol test todo. Remove max_steps trainer arg from cli_main.

  matsumotosan

  committed
  7e58019

  Copy full SHA for 7e58019

Commits on Sep 28, 2022

• Fix max_epoch argparse conflict with trainer argparse method.

  matsumotosan

  committed
  efeaec5

  Copy full SHA for efeaec5