viz_square by default normalize data, that causes corruption of original weights
viz_square by default normalize data, that causes corruption of original weights