zilunpeng/set_teacher_wav2vec2.py

Created March 23, 2021 21:15

Star () You must be signed in to star a gist
Fork () You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/zilunpeng/5ae5ce2eb8cab1a634ef4d6ff06ce00f.js"></script>
Save zilunpeng/5ae5ce2eb8cab1a634ef4d6ff06ce00f to your computer and use it in GitHub Desktop.

Download ZIP

Set the teacher model to evaluation mode. Code below is part of the knowledge distillation toolkit (https://git.io/JYePf).

Raw

set_teacher_wav2vec2.py

self.teacher_model.eval()

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment