transformers/tests/wav2vec2
Sanchit Gandhi 1a62b25caf
Backprop Test for Freeze FlaxWav2Vec2 Feature Encoder (#15938)
* Backprop Test for Freeze FlaxWav2Vec2 Feature Encoder

* remove jnp.ndarray type suggestion

* assert frozen grads are precisely zero
2022-03-07 18:10:15 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_feature_extraction_wav2vec2.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_flax_wav2vec2.py Backprop Test for Freeze FlaxWav2Vec2 Feature Encoder (#15938) 2022-03-07 18:10:15 +01:00
test_modeling_tf_wav2vec2.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_modeling_wav2vec2.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_processor_wav2vec2.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_tokenization_wav2vec2.py Fixing the timestamps with chunking. (#15843) 2022-02-28 21:00:21 +01:00