Code

StormCast training code improvements#738

Code

Merged

pzharrington merged 26 commits intoNVIDIA:mainNVIDIA/modulus:mainfrom pzharrington:stormcast-nn-consolpzharrington/modulus:stormcast-nn-consolCopy head branch name to clipboardDec 16, 2024

+1,677-2,941

Commits

Commits on Nov 12, 2024

• adding stormcast raw files

  

  nvssh nssswitch user account

  committed
  d08ab3f

  Copy full SHA for d08ab3f

Commits on Nov 14, 2024

• major cleanup, refactor and consolidation

  Show description for f2fdad8
  pzharrington

  committed
  f2fdad8

  Copy full SHA for f2fdad8

Commits on Nov 15, 2024

• More cleanup and init readme

  Show description for 999b9d7
  pzharrington

  committed
  999b9d7

  Copy full SHA for 999b9d7

• port command line args to standard argparse

  Show description for 145f01d
  pzharrington

  committed
  145f01d

  Copy full SHA for 145f01d

• remove unused network and loss wrappers

  Show description for 2c161d8
  pzharrington

  committed
  2c161d8

  Copy full SHA for 2c161d8

• add torchrun instructions

  Show description for 546752a
  pzharrington

  committed
  546752a

  Copy full SHA for 546752a

• drop dnnlib utils

  Show description for c69c84b
  pzharrington

  committed
  c69c84b

  Copy full SHA for c69c84b

• use Modulus DistributedManager, streamline cmd args

  Show description for 6babca4
  pzharrington

  committed
  6babca4

  Copy full SHA for 6babca4

Commits on Nov 16, 2024

• Use standard torch checkpoints instead of pickles

  Show description for bc21616
  pzharrington

  committed
  bc21616

  Copy full SHA for bc21616

• Standardize model configs and channel selection across training and inference

  Show description for ca3fbe8
  pzharrington

  committed
  ca3fbe8

  Copy full SHA for ca3fbe8

Commits on Nov 18, 2024

• checkpoint format standardization for train/inference

  Show description for cc3919a
  pzharrington

  committed
  cc3919a

  Copy full SHA for cc3919a

• finalize additional deps

  Show description for 9cd1c1a
  pzharrington

  committed
  9cd1c1a

  Copy full SHA for 9cd1c1a

• format and linting

  Show description for 33cb8a3
  pzharrington

  committed
  33cb8a3

  Copy full SHA for 33cb8a3

Commits on Nov 19, 2024

• drop docker and update changelog

  Show description for a9d3c12
  pzharrington

  committed
  a9d3c12

  Copy full SHA for a9d3c12

• Address feedback

  Show description for 181dfec
  pzharrington

  committed
  181dfec

  Copy full SHA for 181dfec

• add variables to readme, rename network types

  Show description for 473a11c
  pzharrington

  committed
  473a11c

  Copy full SHA for 473a11c

Commits on Nov 22, 2024

• swap stormcast to modulus nn and loss defs

  Show description for bf8e416
  pzharrington

  committed
  bf8e416

  Copy full SHA for bf8e416

Commits on Nov 25, 2024

• Swap to modulus checkpoint save and load utils

  Show description for 820b96b
  pzharrington

  committed
  820b96b

  Copy full SHA for 820b96b

Commits on Nov 27, 2024

• Swap to modulus networks/losses, use modulus checkpointing and logging

  Show description for 4e820b8
  pzharrington

  committed
  4e820b8

  Copy full SHA for 4e820b8

Commits on Dec 3, 2024

• add power spectrum to modulus metrics, remove unused utils

  Show description for 59b9610
  pzharrington

  committed
  59b9610

  Copy full SHA for 59b9610

Commits on Dec 5, 2024

• Readme update and unit tests

  Show description for cbcd9ff
  pzharrington

  committed
  cbcd9ff

  Copy full SHA for cbcd9ff

Commits on Dec 6, 2024

• Merge branch 'main' into stormcast-nn-consol

  Show description for 1704738
  pzharrington

  committed
  1704738

  Copy full SHA for 1704738

Commits on Dec 9, 2024

• drop unused files

  Show description for 03f5ba4
  pzharrington

  committed
  03f5ba4

  Copy full SHA for 03f5ba4

• drop unused diffusions files

  Show description for f18f862
  pzharrington

  committed
  f18f862

  Copy full SHA for f18f862

Commits on Dec 13, 2024

• Merge branch 'main' into stormcast-nn-consol

  pzharrington

  committed
  68cca1d

  Copy full SHA for 68cca1d

• update changelog

  Show description for 6875b38
  pzharrington

  committed
  6875b38

  Copy full SHA for 6875b38