Pangu

PANGU

ArchesWeather

        classDiagram
    class ArchesWeather {
        +bool onnx_supported
        +tuple supported_num_spatial_dims
        +bool features_last
        +bool register
        +forward() tuple
    }
    class BaseModel {
    }
    <<abstract>> BaseModel
    BaseModel <|-- ArchesWeather : herits
    class ModelABC {
        +bool register
        +int in_channels
        +int out_channels
        +tuple input_shape
        +check_required_attributes() None
    }
    <<abstract>> ModelABC
    ModelABC <|-- ArchesWeather : herits
    class Module
    <<abstract>> Module
    Module <|-- ArchesWeather : herits
    

ArchesWeather

ArchesWeather model as described in http://arxiv.org/abs/2405.14527.

PanguWeather

        classDiagram
    class PanguWeather {
        +bool onnx_supported
        +Tuple supported_num_spatial_dims
        +bool features_last
        +bool register
        +forward() Tuple
    }
    class BaseModel {
    }
    <<abstract>> BaseModel
    BaseModel <|-- PanguWeather : herits
    class ModelABC {
        +bool register
        +int in_channels
        +int out_channels
        +tuple input_shape
        +check_required_attributes() None
    }
    <<abstract>> ModelABC
    ModelABC <|-- PanguWeather : herits
    class Module
    <<abstract>> Module
    Module <|-- PanguWeather : herits
    

PanguWeather

PanguWeather network as described in http://arxiv.org/abs/2211.02556 and https://www.nature.com/articles/s41586-023-06185-3.