upt_config¶

Classes¶

Configuration for a Transolver model.

Module Contents¶

class upt_config.UPTConfig(/, **data)¶

Bases: pydantic.BaseModel

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)

num_heads: int = None¶

hidden_dim: int = None¶

mlp_expansion_factor: int = None¶

encoder_depth: int = None¶

use_rope: bool = None¶

supernode_pooling_config: emmi.schemas.modules.encoder.SupernodePoolingConfig¶

approximator_config: emmi.schemas.modules.blocks.TransformerBlockConfig¶

decoder_config: emmi.schemas.modules.decoder.DeepPerceiverDecoderConfig¶

bias_layers: bool = None¶