Code

Add SimCLR trainer#1252

Code

Merged

calebrob6 merged 67 commits intomicrosoft:mainmicrosoft/torchgeo:mainfrom isaaccorley:trainers/simclrisaaccorley/torchgeo:trainers/simclrCopy head branch name to clipboardMay 11, 2023

+535-4

Commits

Commits on Apr 15, 2023

• add simclr and tests

  isaaccorley

  committed
  0d5279d

  Copy full SHA for 0d5279d

• add lightly to reqs

  isaaccorley

  committed
  490b863

  Copy full SHA for 490b863

Commits on Apr 16, 2023

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  a7d6e71

  Copy full SHA for a7d6e71

• pyupgrade

  adamjstewart

  committed
  1dabbcd

  Copy full SHA for 1dabbcd

• Copy things from prior implementation

  adamjstewart

  committed
  42f3be0

  Copy full SHA for 42f3be0

• Add SimCLR v2 projection head

  adamjstewart

  committed
  554aa92

  Copy full SHA for 554aa92

• Remove kwargs

  adamjstewart

  committed
  900d378

  Copy full SHA for 900d378

• Call __init__ explicitly

  adamjstewart

  committed
  b580f19

  Copy full SHA for b580f19

• Fix mypy and docs

  adamjstewart

  committed
  5a4408f

  Copy full SHA for 5a4408f

• Can't test newer setuptools

  adamjstewart

  committed
  cda57df

  Copy full SHA for cda57df

• Default to output dim of model

  adamjstewart

  committed
  5bec352

  Copy full SHA for 5bec352

• Add memory bank

  adamjstewart

  committed
  b1a51f8

  Copy full SHA for b1a51f8

• Ignore erroneous warning

  adamjstewart

  committed
  4b1a15d

  Copy full SHA for 4b1a15d

• Fix configs, test SSL4EO

  adamjstewart

  committed
  845519d

  Copy full SHA for 845519d

• Fix a few layer bugs

  adamjstewart

  committed
  776071e

  Copy full SHA for 776071e

• mypy fixes

  adamjstewart

  committed
  7b86686

  Copy full SHA for 7b86686

• kernel_size must be an integer

  adamjstewart

  committed
  3fba197

  Copy full SHA for 3fba197

• Fix SeCo in_channels

  adamjstewart

  committed
  3b48ed5

  Copy full SHA for 3b48ed5

• Get more coverage

  adamjstewart

  committed
  19c5052

  Copy full SHA for 19c5052

• Bump min lightly

  adamjstewart

  committed
  395fe37

  Copy full SHA for 395fe37

• Default logging

  adamjstewart

  committed
  1edc96a

  Copy full SHA for 1edc96a

• Test weights

  adamjstewart

  committed
  eeb5af9

  Copy full SHA for eeb5af9

• mypy fix

  adamjstewart

  committed
  45b3dc6

  Copy full SHA for 45b3dc6

Commits on Apr 17, 2023

• Grab max_epochs from the trainer

  adamjstewart

  committed
  be13c5e

  Copy full SHA for be13c5e

• max_epochs param removed

  adamjstewart

  committed
  1c32651

  Copy full SHA for 1c32651

Commits on Apr 18, 2023

• Use num_features

  adamjstewart

  committed
  0a65af4

  Copy full SHA for 0a65af4

• Remove classification head

  adamjstewart

  committed
  8a4a2c2

  Copy full SHA for 8a4a2c2

• SimCLR uses LARS, with Adam as a backup

  adamjstewart

  committed
  077f0ed

  Copy full SHA for 077f0ed

• Add warnings

  adamjstewart

  committed
  8dc3463

  Copy full SHA for 8dc3463

• Grab num features directly from model

  adamjstewart

  committed
  4082ee3

  Copy full SHA for 4082ee3

• Check if identity

  adamjstewart

  committed
  9e2b54d

  Copy full SHA for 9e2b54d

Commits on Apr 19, 2023

• Match timm model design

  adamjstewart

  committed
  fb78924

  Copy full SHA for fb78924

• Capture warnings

  adamjstewart

  committed
  e07a2c9

  Copy full SHA for e07a2c9

• Fix tests

  adamjstewart

  committed
  f100aed

  Copy full SHA for f100aed

• Increase coverage

  adamjstewart

  committed
  0421728

  Copy full SHA for 0421728

• Fix method name

  adamjstewart

  committed
  bf36410

  Copy full SHA for bf36410

• More typos

  adamjstewart

  committed
  335d9af

  Copy full SHA for 335d9af

• Escape regex

  adamjstewart

  committed
  1e90095

  Copy full SHA for 1e90095

Commits on Apr 20, 2023

• Newer setuptools now supported

  adamjstewart

  committed
  6a04435

  Copy full SHA for 6a04435

• New batch norm for every layer

  adamjstewart

  committed
  c37c510

  Copy full SHA for c37c510

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  93325e5

  Copy full SHA for 93325e5

Commits on Apr 21, 2023

• Merge branch 'main' into trainers/simclr

  adamjstewart

  authored
  2549d51

  Copy full SHA for 2549d51

Commits on Apr 22, 2023

• Merge branch 'main' into trainers/simclr

  isaaccorley

  authored
  d874b17

  Copy full SHA for d874b17

Commits on Apr 23, 2023

• Rename forward arg

  adamjstewart

  committed
  dbaeb3f

  Copy full SHA for dbaeb3f

• Clarify usage of weights parameter

  Show description for ce2de2a
  

  adamjstewart

  and

  calebrob6

  authored
  ce2de2a

  Copy full SHA for ce2de2a

• Fix flake8

  adamjstewart

  committed
  ab677f3

  Copy full SHA for ab677f3

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  8491a19

  Copy full SHA for 8491a19

• Check it

  calebrob6

  committed
  41ba0ff

  Copy full SHA for 41ba0ff

Commits on Apr 24, 2023

• Use hydra

  adamjstewart

  committed
  7da418e

  Copy full SHA for 7da418e

• Track average L2 normed stdev over features

  calebrob6

  committed
  e0040e0

  Copy full SHA for e0040e0

• Merge branch 'trainers/simclr' of github.com:isaaccorley/torchgeo into trainers/simclr

  calebrob6

  committed
  a76fda5

  Copy full SHA for a76fda5

• SimCLR decays lr to 0

  adamjstewart

  committed
  2f838da

  Copy full SHA for 2f838da

• Add lr warmup

  adamjstewart

  committed
  3e9fb28

  Copy full SHA for 3e9fb28

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  ba44bc9

  Copy full SHA for ba44bc9

Commits on Apr 25, 2023

• Fix version access

  adamjstewart

  committed
  04520cb

  Copy full SHA for 04520cb

• Fix LinearLR

  adamjstewart

  committed
  f0e18c7

  Copy full SHA for f0e18c7

• isinstance supports tuples

  adamjstewart

  committed
  594babb

  Copy full SHA for 594babb

• Comment capitalization

  adamjstewart

  committed
  8aa26fa

  Copy full SHA for 8aa26fa

• Require lightly 1.4.3+

  adamjstewart

  committed
  5bc9c6d

  Copy full SHA for 5bc9c6d

• Require lightly 1.4.3+

  adamjstewart

  committed
  e74f55e

  Copy full SHA for e74f55e

Commits on May 3, 2023

• Bump lightly version

  adamjstewart

  committed
  a93cc23

  Copy full SHA for a93cc23

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  a42902d

  Copy full SHA for a42902d

• Add RandomGrayscale

  adamjstewart

  committed
  27b823d

  Copy full SHA for 27b823d

• Flake8 fixes

  adamjstewart

  committed
  eb7b912

  Copy full SHA for eb7b912

• Placate pydocstyle

  adamjstewart

  committed
  19f6935

  Copy full SHA for 19f6935

• Clarify docs

  adamjstewart

  committed
  3b68504

  Copy full SHA for 3b68504

• Pass correct weights

  adamjstewart

  committed
  c3349c1

  Copy full SHA for c3349c1