Learn batch norm, layer norm, weight norm