transolver_config

Classes

TransolverConfig	Configuration for a Transolver model.
TransolverPlusPlusConfig	Configuration for a Transolver++ model.

Module Contents

class transolver_config.TransolverConfig(/, **data)

Bases: emmi.schemas.models.architectures.transformer_config.TransformerConfig

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)

attention_constructor: type

Constructor of the attention module. Defaults to DotProductAttention..

attention_arguments: dict

Additional arguments for the attention module that are only needed for a specific attention impelentation.

class transolver_config.TransolverPlusPlusConfig(/, **data)

Bases: TransolverConfig

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)

attention_constructor: type

Constructor of the attention module. Defaults to DotProductAttention..

attention_arguments: dict

Additional arguments for the attention module that are only needed for a specific attention impelentation.