zilunpeng/get_student_wav2vec2_log_prob.py

Created March 23, 2021 21:18

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/53a7a2cc53de0265686edb34c479b33f.js"></script>
Save zilunpeng/53a7a2cc53de0265686edb34c479b33f to your computer and use it in GitHub Desktop.

Download ZIP

Get log probability of student model. Code below is part of the knowledge distillation toolkit (https://git.io/JYePf).

Raw

get_student_wav2vec2_log_prob.py

	student_net_output = self.student_model(*batch)
	student_log_prob = student_net_output["log_prob"]

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