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

• add lightly to reqs

  isaaccorley

  committed
  490b863

Commits on Apr 16, 2023

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  a7d6e71

• pyupgrade

  adamjstewart

  committed
  1dabbcd

• Copy things from prior implementation

  adamjstewart

  committed
  42f3be0

• Add SimCLR v2 projection head

  adamjstewart

  committed
  554aa92

• Remove kwargs

  adamjstewart

  committed
  900d378

• Call __init__ explicitly

  adamjstewart

  committed
  b580f19

• Fix mypy and docs

  adamjstewart

  committed
  5a4408f

• Can't test newer setuptools

  adamjstewart

  committed
  cda57df

• Default to output dim of model

  adamjstewart

  committed
  5bec352

• Add memory bank

  adamjstewart

  committed
  b1a51f8

• Ignore erroneous warning

  adamjstewart

  committed
  4b1a15d

• Fix configs, test SSL4EO

  adamjstewart

  committed
  845519d

• Fix a few layer bugs

  adamjstewart

  committed
  776071e

• mypy fixes

  adamjstewart

  committed
  7b86686

• kernel_size must be an integer

  adamjstewart

  committed
  3fba197

• Fix SeCo in_channels

  adamjstewart

  committed
  3b48ed5

• Get more coverage

  adamjstewart

  committed
  19c5052

• Bump min lightly

  adamjstewart

  committed
  395fe37

• Default logging

  adamjstewart

  committed
  1edc96a

• Test weights

  adamjstewart

  committed
  eeb5af9

• mypy fix

  adamjstewart

  committed
  45b3dc6

Commits on Apr 17, 2023

• Grab max_epochs from the trainer

  adamjstewart

  committed
  be13c5e

• max_epochs param removed

  adamjstewart

  committed
  1c32651

Commits on Apr 18, 2023

• Use num_features

  adamjstewart

  committed
  0a65af4

• Remove classification head

  adamjstewart

  committed
  8a4a2c2

• SimCLR uses LARS, with Adam as a backup

  adamjstewart

  committed
  077f0ed

• Add warnings

  adamjstewart

  committed
  8dc3463

• Grab num features directly from model

  adamjstewart

  committed
  4082ee3

• Check if identity

  adamjstewart

  committed
  9e2b54d

Commits on Apr 19, 2023

• Match timm model design

  adamjstewart

  committed
  fb78924

• Capture warnings

  adamjstewart

  committed
  e07a2c9

• Fix tests

  adamjstewart

  committed
  f100aed

• Increase coverage

  adamjstewart

  committed
  0421728

• Fix method name

  adamjstewart

  committed
  bf36410

• More typos

  adamjstewart

  committed
  335d9af

• Escape regex

  adamjstewart

  committed
  1e90095

Commits on Apr 20, 2023

• Newer setuptools now supported

  adamjstewart

  committed
  6a04435

• New batch norm for every layer

  adamjstewart

  committed
  c37c510

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  93325e5

Commits on Apr 21, 2023

• Merge branch 'main' into trainers/simclr

  adamjstewart

  authored
  2549d51

Commits on Apr 22, 2023

• Merge branch 'main' into trainers/simclr

  isaaccorley

  authored
  d874b17

Commits on Apr 23, 2023

• Rename forward arg

  adamjstewart

  committed
  dbaeb3f

• Clarify usage of weights parameter

  Show description for ce2de2a
  

  adamjstewart

  and

  calebrob6

  authored
  ce2de2a

• Fix flake8

  adamjstewart

  committed
  ab677f3

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  8491a19

• Check it

  calebrob6

  committed
  41ba0ff

Commits on Apr 24, 2023

• Use hydra

  adamjstewart

  committed
  7da418e

• Track average L2 normed stdev over features

  calebrob6

  committed
  e0040e0

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

  calebrob6

  committed
  a76fda5

• SimCLR decays lr to 0

  adamjstewart

  committed
  2f838da

• Add lr warmup

  adamjstewart

  committed
  3e9fb28

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  ba44bc9

Commits on Apr 25, 2023

• Fix version access

  adamjstewart

  committed
  04520cb

• Fix LinearLR

  adamjstewart

  committed
  f0e18c7

• isinstance supports tuples

  adamjstewart

  committed
  594babb

• Comment capitalization

  adamjstewart

  committed
  8aa26fa

• Require lightly 1.4.3+

  adamjstewart

  committed
  5bc9c6d

• Require lightly 1.4.3+

  adamjstewart

  committed
  e74f55e

Commits on May 3, 2023

• Bump lightly version

  adamjstewart

  committed
  a93cc23

• Merge branch 'main' into trainers/simclr

  adamjstewart

  committed
  a42902d

• Add RandomGrayscale

  adamjstewart

  committed
  27b823d

• Flake8 fixes

  adamjstewart

  committed
  eb7b912

• Placate pydocstyle

  adamjstewart

  committed
  19f6935

• Clarify docs

  adamjstewart

  committed
  3b68504

• Pass correct weights

  adamjstewart

  committed
  c3349c1