SolarRadiationAFNO6H

GlobalMRF202440 GBNVIDIAPyTorch

Import path: earth2studio.models.dx.SolarRadiationAFNO6H

View source on GitHub View install commands

Documentation

Bases: SolarRadiationAFNO

Solar Radiation AFNO diagnostic model for 6-hour frequency predictions. Predicts the accumulated global surface solar radiation over 6 hours [Jm^-2].

__call__

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

Forward pass of diagnostic

load_default_package classmethod

load_default_package() -> Package

Load prognostic package

load_model classmethod

load_model(package: Package) -> DiagnosticModel

Load prognostic from package

Parameters:

• package (Package) –

  Package to load model from

Returns:

• PrognosticModel –

  Prognostic model