Code

Match Nature Paper Implementation#5

Code

Merged

jacobbieker merged 64 commits intomainopenclimatefix/skillful_nowcasting:mainfrom jacob/nature-paperopenclimatefix/skillful_nowcasting:jacob/nature-paperCopy head branch name to clipboardNov 11, 2021

+1,740-702

Commits

Commits on Sep 30, 2021

• Update DBlock to match new paper

  jacobbieker

  committed
  7b36103

  Copy full SHA for 7b36103

• Update LBlock and LatentConditioningStack to match paper better

  jacobbieker

  committed
  a2d9bd3

  Copy full SHA for a2d9bd3

• Add 1x1 channel upsample in generator block

  jacobbieker

  committed
  76255f2

  Copy full SHA for 76255f2

Commits on Oct 5, 2021

• Switch to Average Pooling, as in model

  Show description for 5213387
  jacobbieker

  committed
  5213387

  Copy full SHA for 5213387

• Add BatchNorm in discriminators

  Show description for ec08355
  jacobbieker

  committed
  ec08355

  Copy full SHA for ec08355

• Change test to match DGMR expected inputs

  jacobbieker

  committed
  2380f39

  Copy full SHA for 2380f39

Commits on Oct 14, 2021

• Remove comment

  jacobbieker

  committed
  ebc08c1

  Copy full SHA for ebc08c1

Commits on Oct 18, 2021

• Fix issues with ConvGRU and shapes of inputs

  jacobbieker

  committed
  02c5cea

  Copy full SHA for 02c5cea

• Comment out linting for now

  jacobbieker

  committed
  c562b2f

  Copy full SHA for c562b2f

Commits on Oct 19, 2021

• Add creating RTD documentation

  jacobbieker

  committed
  08d3682

  Copy full SHA for 08d3682

Commits on Nov 8, 2021

• Merge branch 'main' into jacob/nature-paper

  jacobbieker

  authored
  c2606fc

  Copy full SHA for c2606fc

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for ba334b5
  pre-commit-ci[bot]

  committed
  ba334b5

  Copy full SHA for ba334b5

• Add UpsampleGBlock, update GBlock to match pseudocode

  Show description for 0efb9a1
  jacobbieker

  committed
  0efb9a1

  Copy full SHA for 0efb9a1

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  8975c07

  Copy full SHA for 8975c07

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 06a66e5
  pre-commit-ci[bot]

  committed
  06a66e5

  Copy full SHA for 06a66e5

• Add some comments on LBlock Latent conditioning stack

  Show description for 6d526ec
  jacobbieker

  committed
  6d526ec

  Copy full SHA for 6d526ec

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  a61d62d

  Copy full SHA for a61d62d

• Add Attention layer based off pseudocode

  jacobbieker

  committed
  6354ca8

  Copy full SHA for 6354ca8

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for ca85104
  pre-commit-ci[bot]

  committed
  ca85104

  Copy full SHA for ca85104

• Swap out ConvGRU implementations

  Show description for 94b4812
  jacobbieker

  committed
  94b4812

  Copy full SHA for 94b4812

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 2c5231a
  pre-commit-ci[bot]

  committed
  2c5231a

  Copy full SHA for 2c5231a

• Update Dblock and Spatial Discriminator

  jacobbieker

  committed
  1423c18

  Copy full SHA for 1423c18

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  6b89052

  Copy full SHA for 6b89052

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for dc981f2
  pre-commit-ci[bot]

  committed
  dc981f2

  Copy full SHA for dc981f2

• Update Temporal Discriminator

  jacobbieker

  committed
  f1b3927

  Copy full SHA for f1b3927

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  bb1d9d4

  Copy full SHA for bb1d9d4

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for e8782c3
  pre-commit-ci[bot]

  committed
  e8782c3

  Copy full SHA for e8782c3

• Add encapsulating discriminator

  jacobbieker

  committed
  d4a896e

  Copy full SHA for d4a896e

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  c9a89e8

  Copy full SHA for c9a89e8

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 1f2fc3c
  pre-commit-ci[bot]

  committed
  1f2fc3c

  Copy full SHA for 1f2fc3c

• Add missing mixing layer to ContextConditioningStack

  Show description for f5204a0
  jacobbieker

  committed
  f5204a0

  Copy full SHA for f5204a0

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  d9c3ff0

  Copy full SHA for d9c3ff0

• Add pseudocode

  jacobbieker

  committed
  913f76e

  Copy full SHA for 913f76e

• Add pseudocode

  jacobbieker

  committed
  3fbbd5c

  Copy full SHA for 3fbbd5c

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 37775ae
  pre-commit-ci[bot]

  committed
  37775ae

  Copy full SHA for 37775ae

• Update sampler and ConvGRU

  jacobbieker

  committed
  b080785

  Copy full SHA for b080785

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 426e7a7
  pre-commit-ci[bot]

  committed
  426e7a7

  Copy full SHA for 426e7a7

• Update sampler and ConvGRU

  jacobbieker

  committed
  4a87775

  Copy full SHA for 4a87775

• Merge remote-tracking branch 'origin/jacob/nature-paper' into jacob/nature-paper

  jacobbieker

  committed
  1ea30f5

  Copy full SHA for 1ea30f5

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for eef675f
  pre-commit-ci[bot]

  committed
  eef675f

  Copy full SHA for eef675f

• Update PL model and training steps to match pseudocode

  jacobbieker

  committed
  81b21c2

  Copy full SHA for 81b21c2

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 556c130
  pre-commit-ci[bot]

  committed
  556c130

  Copy full SHA for 556c130

• Fix 'valid' typing

  jacobbieker

  committed
  9deeffb

  Copy full SHA for 9deeffb

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 83072cd
  pre-commit-ci[bot]

  committed
  83072cd

  Copy full SHA for 83072cd

Commits on Nov 9, 2021

• Add spectral norm to ConvGRU Conv2D

  jacobbieker

  committed
  4546a3c

  Copy full SHA for 4546a3c

• Better match mixing layer

  jacobbieker

  committed
  8165e61

  Copy full SHA for 8165e61

• Fix LBlock

  jacobbieker

  committed
  6e20bed

  Copy full SHA for 6e20bed

• Fix LatentConditioningStack

  jacobbieker

  committed
  baedd53

  Copy full SHA for baedd53

• Fix latent dim shape

  jacobbieker

  committed
  60bd1bb

  Copy full SHA for 60bd1bb

• Fix UpsampleGBlock

  jacobbieker

  committed
  00ab251

  Copy full SHA for 00ab251

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 6d94b34
  pre-commit-ci[bot]

  committed
  6d94b34

  Copy full SHA for 6d94b34

• Spectrally normalize most convolutions

  jacobbieker

  committed
  a7307df

  Copy full SHA for a7307df

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 7295d7c
  pre-commit-ci[bot]

  committed
  7295d7c

  Copy full SHA for 7295d7c

• Add discriminator tests

  jacobbieker

  committed
  064f17c

  Copy full SHA for 064f17c

• Add discriminator tests

  jacobbieker

  committed
  1df80bc

  Copy full SHA for 1df80bc

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 26b82f0
  pre-commit-ci[bot]

  committed
  26b82f0

  Copy full SHA for 26b82f0

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for edcf549
  pre-commit-ci[bot]

  committed
  edcf549

  Copy full SHA for edcf549

• Auto stash before merge of "jacob/nature-paper" and "origin/jacob/nature-paper"

  jacobbieker

  committed
  de55aba

  Copy full SHA for de55aba

• Add ConvGRU test

  jacobbieker

  committed
  548f2ee

  Copy full SHA for 548f2ee

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for e4ae119
  pre-commit-ci[bot]

  committed
  e4ae119

  Copy full SHA for e4ae119

• Add failing test for every step in Sampler

  Show description for 0f517c5
  jacobbieker

  committed
  0f517c5

  Copy full SHA for 0f517c5

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 7d4c987
  pre-commit-ci[bot]

  committed
  7d4c987

  Copy full SHA for 7d4c987

• Add missing spectral norm

  jacobbieker

  committed
  8520569

  Copy full SHA for 8520569

• [pre-commit.ci] auto fixes from pre-commit.com hooks

  Show description for 8af6638
  pre-commit-ci[bot]

  committed
  8af6638

  Copy full SHA for 8af6638