/wxbtool

A toolkit for WeatherBench based on PyTorch

Primary LanguagePythonMIT LicenseMIT

wxbtool

DOI

A toolkit for WeatherBench based on PyTorch

Install

pip install wxbtool

Cheat sheet

Start a data set server for 3-days prediction of t850 by Weyn's solution

wxb dserve -m wxbtool.specs.res5_625.t850weyn -s Setting3d

Start a training process for a UNet model following Weyn's solution

wxb train -m wxbtool.zoo.res5_625.unet.t850d3sm_weyn

Start a testing process for a UNet model following Weyn's solution

wxb test -m wxbtool.zoo.res5_625.unet.t850d3sm_weyn

Start a data set server with http binding

wxb dserve -m wxbtool.specs.res5_625.t850weyn -s Setting3d

Start a data set server with ip and port

wxb dserve -m wxbtool.specs.res5_625.t850weyn -s Setting3d -b 127.0.0.1:8088

Start a training process with unix socket binding

wxb train -m wxbtool.zoo.res5_625.unet.t850d3sm_weyn -d unix:/tmp/test.sock

How to use

  • quick start
  • understanding the physical process by plotting
  • develop your own neural model
  • try a toy physical model
  • explore the possibility to combine neural and physical model together

How to release

uv build
uv publish

git tag va.b.c master
git push origin va.b.c

Contributors