benchmarl.models.CnnConfig

class CnnConfig(cnn_num_cells: Sequence[int], cnn_kernel_sizes: Sequence[int] | int, cnn_strides: Sequence[int] | int, cnn_paddings: Sequence[int] | int, cnn_activation_class: Type[Module], mlp_num_cells: Sequence[int], mlp_layer_class: Type[Module], mlp_activation_class: Type[Module], cnn_activation_kwargs: dict | None = None, cnn_norm_class: Type[Module] | None = None, cnn_norm_kwargs: dict | None = None, mlp_activation_kwargs: dict | None = None, mlp_norm_class: Type[Module] | None = None, mlp_norm_kwargs: dict | None = None)

Bases: ModelConfig

Dataclass config for a Cnn.

static associated_class()

The associated Model class