ecmwf

/

aifs-single-0.2.1

like 26

Follow

ECMWF 96

• "anemoi.models.layers.conv.GraphTransformerConv",
• "torch.nn.modules.container.Sequential",
• "anemoi.models.layers.processor.TransformerProcessor",
• "_operator.getitem",
• "__builtin__.getattr",
• "numpy.dtype",
• "torch_geometric.data.graph_store.EdgeAttr",
• "torch_geometric.data.storage.BaseStorage",
• "_codecs.encode",
• "pathlib.PosixPath",
• "anemoi.models.layers.chunk.TransformerProcessorChunk",
• "anemoi.models.models.encoder_processor_decoder.AnemoiModelEncProcDec",
• "anemoi.models.preprocessing.normalizer.InputNormalizer",
• "anemoi.models.layers.attention.MultiHeadSelfAttention",
• "torch.IntStorage",
• "anemoi.models.data_indices.collection.IndexCollection",
• "__builtin__.set",
• "anemoi.models.data_indices.tensor.OutputTensorIndex",
• "__builtin__.long",
• "anemoi.models.layers.block.GraphTransformerMapperBlock",
• "inspect._ParameterKind",
• "torch.nn.modules.linear.Linear",
• "torch.LongStorage",
• "torch.DoubleStorage",
• "anemoi.models.preprocessing.Processors",
• "torch_geometric.data.hetero_data.HeteroData",
• "torch_geometric.nn.aggr.basic.SumAggregation",
• "inspect._empty",
• "torch._utils._rebuild_tensor_v2",
• "__builtin__.type",
• "flash_attn.flash_attn_interface.flash_attn_func",
• "numpy.ndarray",
• "anemoi.models.layers.mapper.GraphTransformerBackwardMapper",
• "anemoi.models.data_indices.index.DataIndex",
• "typing.Union",
• "anemoi.models.interface.AnemoiModelInterface",
• "anemoi.models.layers.mapper.GraphTransformerForwardMapper",
• "torch.nn.modules.normalization.LayerNorm",
• "torch_geometric.data.storage.EdgeStorage",
• "torch_geometric.data.storage.NodeStorage",
• "collections.OrderedDict",
• "torch.FloatStorage",
• "torch._utils._rebuild_parameter",
• "torch.Tensor",
• "anemoi.models.layers.block.TransformerProcessorBlock",
• "torch.nn.modules.container.ModuleDict",
• "anemoi.models.data_indices.index.ModelIndex",
• "torch_geometric.data.feature_store.TensorAttr",
• "anemoi.models.layers.graph.TrainableTensor",
• "torch_geometric.nn.conv.utils.inspector.Inspector",
• "torch.nn.modules.activation.GELU",
• "torch.nn.modules.container.ModuleList",
• "anemoi.models.data_indices.tensor.InputTensorIndex",
• "inspect.Parameter",
• "numpy.core.multiarray._reconstruct"

How to fix it?