ecmwf

/

aifs-ens-2.0

like 0

Follow

ECMWF 137

• "anemoi.models.layers.bounding.FractionBounding",
• "torch_geometric.data.storage.NodeStorage",
• "collections.defaultdict",
• "pathlib.PosixPath",
• "torch_geometric.data.storage.BaseStorage",
• "torch_geometric.data.storage.EdgeStorage",
• "__builtin__.list",
• "anemoi.models.layers.attention.FlashAttentionWrapper",
• "anemoi.models.layers.sparse_projector.SparseProjector",
• "anemoi.models.layers.bounding.ReluBounding",
• "omegaconf.base.ContainerMetadata",
• "typing.Any",
• "anemoi.models.preprocessing.postprocessor.ConditionalNaNPostprocessor",
• "torch.nn.modules.container.ModuleList",
• "anemoi.models.preprocessing.Processors",
• "anemoi.models.preprocessing.normalizer.InputNormalizer",
• "anemoi.models.layers.block.GraphTransformerMapperBlock",
• "torch._utils._rebuild_tensor_v2",
• "torch.BoolStorage",
• "anemoi.models.layers.bounding.HardtanhBounding",
• "anemoi.models.layers.mapper.GraphTransformerForwardMapper",
• "anemoi.models.layers.graph.NamedNodesAttributes",
• "numpy.core.multiarray._reconstruct",
• "flash_attn.flash_attn_interface.flash_attn_func",
• "__builtin__.getattr",
• "torch.serialization._get_layout",
• "anemoi.models.layers.graph.TrainableTensor",
• "torch._utils._rebuild_parameter",
• "__builtin__.long",
• "anemoi.models.data_indices.index.ModelIndex",
• "torch_geometric.data.graph_store.EdgeAttr",
• "anemoi.models.data_indices.tensor.OutputTensorIndex",
• "anemoi.models.triton.gt.GraphTransformerFunction",
• "anemoi.models.layers.attention.MultiHeadSelfAttention",
• "anemoi.models.interface.AnemoiModelInterface",
• "anemoi.models.layers.mapper.GraphTransformerBackwardMapper",
• "torch._utils._rebuild_sparse_tensor",
• "numpy.dtype",
• "torch.LongStorage",
• "omegaconf.listconfig.ListConfig",
• "anemoi.models.models.ens_encoder_processor_decoder.AnemoiEnsModelEncProcDec",
• "torch.nn.modules.activation.GELU",
• "torch._utils._rebuild_parameter_with_state",
• "anemoi.models.layers.residual.SkipConnection",
• "anemoi.models.layers.block.TransformerProcessorBlock",
• "numpy.ndarray",
• "anemoi.models.layers.ensemble.NoiseConditioning",
• "__builtin__.dict",
• "torch.nn.modules.container.Sequential",
• "anemoi.utils.config.DotDict",
• "anemoi.models.data_indices.collection.IndexCollection",
• "anemoi.models.layers.normalization.AutocastLayerNorm",
• "torch_geometric.data.hetero_data.HeteroData",
• "anemoi.models.data_indices.index.DataIndex",
• "torch.Size",
• "anemoi.models.data_indices.tensor.InputTensorIndex",
• "functools.partial",
• "_codecs.encode",
• "torch.IntStorage",
• "torch.nn.modules.linear.Linear",
• "anemoi.models.layers.normalization.ConditionalLayerNorm",
• "omegaconf.base.Metadata",
• "torch.nn.modules.container.ModuleDict",
• "omegaconf.dictconfig.DictConfig",
• "torch.nn.modules.normalization.LayerNorm",
• "anemoi.models.layers.mlp.MLP",
• "collections.OrderedDict",
• "torch_geometric.data.feature_store.TensorAttr",
• "anemoi.models.layers.processor.TransformerProcessor",
• "omegaconf.nodes.AnyNode",
• "torch.FloatStorage",
• "anemoi.models.preprocessing.imputer.ConstantImputer",
• "torch.nn.modules.linear.Identity",
• "__builtin__.set"

How to fix it?