noether.core.schemas.models.upt¶

Classes¶

Configuration for a Transolver model.

Module Contents¶

class noether.core.schemas.models.upt.UPTConfig(/, **data)¶

Bases: noether.core.schemas.models.base.ModelBaseConfig

Configuration for a Transolver model.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:: data (Any)

model_config¶: Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

num_heads: int = None¶: Number of attention heads in the model.

hidden_dim: int = None¶: Hidden dimension of the model.

mlp_expansion_factor: int = None¶: Expansion factor for the MLP of the FF layers.

approximator_depth: int = None¶: Number of approximator layers.

use_rope: bool = None¶

supernode_pooling_config: noether.core.schemas.modules.SupernodePoolingConfig¶

approximator_config: noether.core.schemas.modules.blocks.TransformerBlockConfig¶

decoder_config: noether.core.schemas.modules.DeepPerceiverDecoderConfig¶

bias_layers: bool = None¶

data_specs: noether.core.schemas.dataset.AeroDataSpecs¶