benchmarl.models.MlpConfig

class MlpConfig(num_cells: Sequence[int], layer_class: Type[Module], activation_class: Type[Module], activation_kwargs: dict | None = None, norm_class: Type[Module] | None = None, norm_kwargs: dict | None = None, num_feature_dims: int = 1)

Bases: ModelConfig

Dataclass config for a Mlp.

static associated_class()

The associated Model class