torchcell.models.hetero_cell_bipartite_dango_gi.get_norm_layer

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

Return a LayerNorm or BatchNorm1d layer for the given norm name.