wxbtool

A toolkit for WeatherBench based on PyTorch

Installation

pip install wxbtool

For detailed installation instructions, see the Installation Guide.

Quick Start

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 an inference process for a UNet model following Weyn's solution

wxb infer -m wxbtool.zoo.res5_625.unet.t850d3sm_weyn -t 2023-01-01T00:00:00 -o output.png

Start a GAN inference process for a UNet model following Weyn's solution

wxb inferg -m wxbtool.zoo.res5_625.unet.t850d3sm_weyn -t 2023-01-01T00:00:00 -s 10 -o output.nc

Start a data set server with http binding

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

Start a training process with unix socket binding

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

For more detailed examples and explanations, see the Quick Start Guide.

Documentation

User Documentation

• Installation Guide
• Quick Start Guide
• Data Handling Guide
• Training Guide
• Evaluation Guide
• Inference Guide
• Troubleshooting Guide

Technical Documentation

• Architecture Overview
• Model Specifications
• Creating Custom Models

How to use

See the comprehensive documentation in the docs directory.

How to release

uv build
uv publish

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

Contributors

• Mingli Yuan (Mountain)
• Ren Lu