tests/test_modeling_flax_wav2vec2.py · 2e9fb13fb10078e9419ea96607eb66a9f46d0ca1 · chenpangpang / transformers

[Wav2Vec2] Correctly pad mask indices for PreTraining (#12748) · 2e9fb13f

Patrick von Platen authored Jul 15, 2021



* fix_torch_device_generate_test

* remove @

* start adding tests

* correct wav2vec2 pretraining

* up

* up
Co-authored-by: Patrick von Platen <patrick@huggingface.co>

2e9fb13f

test_modeling_flax_wav2vec2.py 16.1 KB

Replace test_modeling_flax_wav2vec2.py