transformers/tests/models/vit_mae
Joshua Lochner 18c5b216f1
Fix ViT-MAE decoder interpolate (#33330)
* Fix ViT-MAE decoder interpolate

* Add unit test for `interpolate_pos_encoding` w/ custom sizes

* [run_slow] vit_mae
2024-09-30 18:47:13 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_vit_mae.py added interpolation for vitmae model in pytorch as well as tf. (#30732) 2024-05-24 16:20:09 +01:00
test_modeling_vit_mae.py Fix ViT-MAE decoder interpolate (#33330) 2024-09-30 18:47:13 +02:00