src/transformers/models/wav2vec2/modeling_wav2vec2.py · 49d62b01783416a89acc0b865f7cb8dbab87cd6b · chenpangpang / transformers

[Wav2Vec2] Fix None loss in doc examples (#19218) · 49d62b01

rbsteinm authored Sep 29, 2022

* pass sampled_negative_indices parameter to the model to avoid getting a None loss
* concerns doc examples for Wav2Vec2ForPreTraining and Wav2Vec2ConformerForPreTraining

49d62b01

modeling_wav2vec2.py 90.6 KB

Replace modeling_wav2vec2.py