How are the pixel means and std calculated? I trained InfoMin on a different dataset and am planning on transferring it to a downstream task. 