torchcell.models.hetero_cell_bipartite_dango_gi_lazy.get_norm_layer

get_norm_layer(channels: int, norm: str) → Module[source]

Get normalization layer from norm_register - NO FALLBACK.

Parameters:

  • channels – Number of channels for normalization

  • norm – Normalization type string (must be in norm_register)

Returns:

Normalization layer instance

Raises:

ValueError – If norm not found in norm_register