torchcell.models.hetero_cell_bipartite_dango.get_norm_layer

get_norm_layer(channels: int, norm: str) → Module

Return a layer or batch normalization module for the given channel count.