noether.core.schemas.modules.layers.scalar_conditioner

Classes

ScalarsConditionerConfig

Module Contents

class noether.core.schemas.modules.layers.scalar_conditioner.ScalarsConditionerConfig(/, **data)

Bases: pydantic.BaseModel

Parameters:

data (Any)

hidden_dim: int = None

Dimension for embedding the scalars and the per-scalar MLP.

num_scalars: int = None

How many scalars are embedded.

condition_dim: int | None = None

Dimension of the final conditioning vector. Defaults to 4 * dim if condition_dim is None.

init_weights: noether.core.types.InitWeightsMode = 'truncnormal002'

Weight initialization for MLPs.