ClimateNet

GlobalMRF202140 GBNVIDIAPyTorch

Import path: earth2studio.models.dx.ClimateNet

View source on GitHub View install commands

Documentation

Bases: Module, AutoModelMixin

Climate Net diagnostic model, built into Earth2Studio. This model can be used to create prediction labels for tropical cyclones and atmospheric rivers from a set of three atmospheric variables on a quater degree resolution equirectangular grid. It produces three non-standard output channels climnet_bg, climnet_tc and climnet_ar representing background label, tropical cyclone and atmospheric river labels.

Note

This model and checkpoint are from Prabhat et al. 2021. For more information see the following references:

• doi.org/10.5194/gmd-14-107-2021
• andregraubner/ClimateNet

Parameters:

• core_model (Module) –

  Core pytorch model

• center (Tensor) –

  Model center normalization tensor of size [20,1,1]

• scale (Tensor) –

  Model scale normalization tensor of size [20,1,1]

__call__

__call__(
    x: Tensor, coords: CoordSystem
) -> tuple[Tensor, CoordSystem]

Forward pass of diagnostic

load_default_package classmethod

load_default_package() -> Package

Default pre-trained climatenet model package from Nvidia model registry

load_model classmethod

load_model(package: Package) -> DiagnosticModel

Load diagnostic from package