Sanchit Gandhi
|
925fc57b70
|
[Flax] Improve Robustness of Back-Prop Tests (#16418)
* [Flax] Improve Robustness of Back-Prop Tests
* check equality of logits/outputs
* make fixup
|
2022-03-28 11:56:54 +02:00 |
|
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 |
|
Sanchit Gandhi
|
3c4fbc616f
|
Freeze FlaxWav2Vec2 Feature Encoder (#15873)
* Freeze FlaxWav2Vec2 Feature Encoder
* add to all module apply
* add backprop test
|
2022-03-03 14:17:13 +01:00 |
|
Sanchit Gandhi
|
8fd4731072
|
Fix Bug in FlaxWav2Vec2 Slow Test (#15887)
|
2022-03-02 16:02:26 +01:00 |
|
Lysandre Debut
|
29c10a41d0
|
[Test refactor 1/5] Per-folder tests reorganization (#15725)
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
|
2022-02-23 15:46:28 -05:00 |
|