torchcell.models.hetero_cell_nsa_retry.get_norm_layer

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

Create a normalization layer based on specified type.