mindearth.cell

API Name

Description

Supported Platforms

mindearth.cell.AFNONet

The AFNO model is a deep learning model that based on the Fourier Neural Operator (AFNO) and the Vision Transformer structure.

Ascend GPU

mindearth.cell.DEMNet

Digital Elevation Model is based on deep residual network and transfer learning.

Ascend GPU

mindearth.cell.DgmrDiscriminator

The Dgmr Discriminator is based on Temporal Discriminator and Spatial Discriminator, which contains deep residual block.

Ascend GPU

mindearth.cell.DgmrGenerator

The Dgmr Generator is based on Conditional_Stack, Latent_Stack, Upsample_Stack and ConvGRU, which contain deep residual block.

Ascend GPU

mindearth.cell.GraphCastNet

The GraphCast is based on graph neural networks and a novel high-resolution multi-scale mesh representation autoregressive model.

Ascend GPU

mindearth.cell.ViTKNO

The ViT-KNO is a deep learning model that based on the Koopman theory and the Vision Transformer structure.

Ascend GPU