Self-Supervised Learning for Aligning Intermediate Layers