tests/wav2vec2/test_modeling_flax_wav2vec2.py · 1a62b25caf06cd4a13af2db1e94abce9969a1d9b · chenpangpang / transformers

Sanchit Gandhi authored Mar 07, 2022

* Backprop Test for Freeze FlaxWav2Vec2 Feature Encoder

* remove jnp.ndarray type suggestion

* assert frozen grads are precisely zero

1a62b25c

test_modeling_flax_wav2vec2.py 22.1 KB