Imported from https://github.com/intersun/PKD-for-BERT-Model-Compression