Allow loading of teacher weights for the student in distillation. If option is set to true, then any variables from the teacher checkpoint that match the naming of the student checkpoint will be loaded.
Allow loading of teacher weights for the student in distillation. If option is set to true, then any variables from the teacher checkpoint that match the naming of the student checkpoint will be loaded.

PiperOrigin-RevId: 330602166

tensorflow/mesh,tensorflow/mesh