Imported from https://github.com/shelling203/SpecAugment