transformers/tests/models/data2vec
Matt 134caef31a
Speed up TF tests by reducing hidden layer counts (#24595)
* hidden layers, huh, what are they good for (absolutely nothing)

* Some tests break with 1 hidden layer, use 2

* Use 1 hidden layer in a few slow models

* Use num_hidden_layers=2 everywhere

* Slightly higher tol for groupvit

* Slightly higher tol for groupvit
2023-06-30 16:30:33 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_data2vec_audio.py Fix TypeError: Object of type int64 is not JSON serializable (#24340) 2023-06-27 12:15:49 +01:00
test_modeling_data2vec_text.py 🔥Rework pipeline testing by removing PipelineTestCaseMeta 🚀 (#21516) 2023-02-28 19:40:57 +01:00
test_modeling_data2vec_vision.py Update old existing feature extractor references (#24552) 2023-06-29 10:17:36 +01:00
test_modeling_tf_data2vec_vision.py Speed up TF tests by reducing hidden layer counts (#24595) 2023-06-30 16:30:33 +01:00