ecmwf

/

aifs-ens-1.0

like 9

Follow

ECMWF 108

• "scipy.sparse._csr.csr_array",
• "torch.FloatStorage",
• "anemoi.models.layers.mapper.GraphTransformerForwardMapper",
• "torch_geometric.data.storage.EdgeStorage",
• "torch_geometric.data.feature_store.TensorAttr",
• "anemoi.models.layers.block.TransformerProcessorBlock",
• "anemoi.models.preprocessing.normalizer.InputNormalizer",
• "__builtin__.dict",
• "numpy.ndarray",
• "anemoi.models.layers.ensemble.NoiseConditioning",
• "anemoi.models.layers.graph.TrainableTensor",
• "anemoi.models.layers.processor.TransformerProcessor",
• "pathlib.PosixPath",
• "inspect.Parameter",
• "__builtin__.list",
• "anemoi.models.interface.AnemoiModelInterface",
• "torch.IntStorage",
• "omegaconf.base.Metadata",
• "numpy._core.multiarray._reconstruct",
• "anemoi.models.layers.attention.FlashAttentionWrapper",
• "torch_geometric.nn.conv.utils.inspector.Inspector",
• "anemoi.models.layers.conv.GraphTransformerConv",
• "torch_geometric.data.storage.NodeStorage",
• "anemoi.models.layers.attention.MultiHeadSelfAttention",
• "typing.Union",
• "anemoi.models.layers.normalization.AutocastLayerNorm",
• "torch.nn.modules.container.ModuleList",
• "numpy.dtype",
• "anemoi.models.layers.mlp.MLP",
• "torch.nn.modules.linear.Linear",
• "torch.Tensor",
• "torch.nn.modules.normalization.LayerNorm",
• "anemoi.models.models.ens_encoder_processor_decoder.AnemoiEnsModelEncProcDec",
• "__builtin__.long",
• "torch.serialization._get_layout",
• "torch.nn.modules.linear.Identity",
• "torch._utils._rebuild_tensor_v2",
• "torch_geometric.data.storage.BaseStorage",
• "anemoi.models.data_indices.tensor.OutputTensorIndex",
• "collections.OrderedDict",
• "omegaconf.nodes.AnyNode",
• "anemoi.models.preprocessing.Processors",
• "__builtin__.set",
• "omegaconf.dictconfig.DictConfig",
• "inspect._ParameterKind",
• "anemoi.models.layers.graph.NamedNodesAttributes",
• "torch_geometric.nn.aggr.basic.SumAggregation",
• "anemoi.models.data_indices.collection.IndexCollection",
• "torch.nn.modules.activation.GELU",
• "torch_geometric.data.graph_store.EdgeAttr",
• "_codecs.encode",
• "torch_geometric.data.hetero_data.HeteroData",
• "anemoi.models.layers.mapper.GraphTransformerBackwardMapper",
• "__builtin__.type",
• "torch._utils._rebuild_sparse_tensor",
• "typing.Any",
• "__builtin__.getattr",
• "torch.nn.modules.container.Sequential",
• "omegaconf.base.ContainerMetadata",
• "anemoi.models.layers.normalization.ConditionalLayerNorm",
• "flash_attn.flash_attn_interface.flash_attn_func",
• "omegaconf.listconfig.ListConfig",
• "anemoi.models.layers.chunk.TransformerProcessorChunk",
• "anemoi.models.data_indices.index.DataIndex",
• "anemoi.models.data_indices.index.ModelIndex",
• "anemoi.models.data_indices.tensor.InputTensorIndex",
• "anemoi.models.layers.block.GraphTransformerMapperBlock",
• "inspect._empty",
• "anemoi.models.layers.bounding.HardtanhBounding",
• "anemoi.models.layers.bounding.ReluBounding",
• "anemoi.models.layers.bounding.FractionBounding",
• "torch.nn.modules.container.ModuleDict",
• "collections.defaultdict",
• "_operator.getitem",
• "anemoi.utils.config.DotDict",
• "torch._utils._rebuild_parameter",
• "torch.LongStorage",
• "torch.Size"

How to fix it?