Imported from https://github.com/huanghonggit/Mask-Language-Model