Imported from https://github.com/RayeRen/multilingual-kd-pytorch