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

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_modeling_flax_speech_encoder_decoder.py

[Flax] Improve Robustness of Back-Prop Tests (#16418)

2022-03-28 11:56:54 +02:00

test_modeling_speech_encoder_decoder.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00