zilunpeng/get_wav2vec2_output.py

Created March 23, 2021 21:07

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

Download ZIP

Get the output from wav2vec 2.0. Code below is part of the wav2vec 2.0 inference notebook (https://git.io/JYeKX).

Raw

get_wav2vec2_output.py

	encoder_out = model(**encoder_input)
	emissions = model.get_normalized_probs(encoder_out, log_probs=True)
	emissions = emissions.transpose(0, 1).float().cpu().contiguous()

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