/sfno_shallowwater

comparison of fourier neural operator (FNO) and spherical fourier neural operator for solving the shallow water equations

Primary LanguagePythonMIT LicenseMIT

Comparison of FNO and SFNO for solving the Shallow Water Equations

this is mainly made to demonstrate how you can combine the following tools

  1. hydra
  2. lightning
  3. wandb and use them efficiently across different hardware setups, e.g. your laptop for debugging and a SLURM cluster for training. You'll find explanations on it on my blog

Launching Training

python train.py --config-name=neuraloperator

Since the CLI uses Hydra, you can override any hyperparams in the config (also see the blog for that).