I want to replace the BatchNormalization layers in a custo keras model with GroupNormalization ones. I have seen a workaround here but it doesn\'t work because I can\'t buil