transformers/tests/models/beit
Alara Dirik 36b9a99433
Fix BeitFeatureExtractor postprocessing (#19119)
* return post-processed segmentations as list, add test
* use torch to resize logits
* fix assertion error if no target_size is specified
2022-09-20 18:53:40 +03:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_beit.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_beit.py Fix BeitFeatureExtractor postprocessing (#19119) 2022-09-20 18:53:40 +03:00
test_modeling_flax_beit.py Improve vision models (#17731) 2022-06-24 11:34:51 +02:00