requirements: torch >= 1.6, transformers >= 3.3.0
Train a teacher model: run train_fst.py in tracher directory.
Train a student model: run override_aug.py in student directory. A teacher model is needed and the model path needs to be specified in override_aug.py.