benchmarl.models.DeepsetsConfig

class DeepsetsConfig(aggr: str, out_features_local_nn: int, local_nn_num_cells: Sequence[int], local_nn_activation_class: Type[Module], global_nn_num_cells: Sequence[int], global_nn_activation_class: Type[Module])

Bases: ModelConfig

Dataclass config for a Deepsets.

static associated_class()

The associated Model class