transformers/tests/models/vit_mae
Joao Gante 658010c739
TF: tests for (de)serializable models with resized tokens (#19013)
* resized models that we can actually load

* separate embeddings check

* add test for embeddings out of bounds

* add fake slows
2022-09-16 16:38:08 +01:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_vit_mae.py TF: tests for (de)serializable models with resized tokens (#19013) 2022-09-16 16:38:08 +01:00
test_modeling_vit_mae.py Improve vision models (#17731) 2022-06-24 11:34:51 +02:00