configs/model/network/mid/mlp_multi.yaml

defaults:
  - activation: gelu
  - norm: identity

instance:
  _target_: models.networks.mlp.MLP
  initial_dim: ${model.network.backbone.output_dim}
  hidden_dim:
    - ${model.network.backbone.output_dim}
    - 64
  final_dim: ${model.network.head.final_dim}
  norm: ${model.network.mid.norm}
  activation: ${model.network.mid.activation}
  aux_data: ${aux_data}