riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity
71e602725b
transformers/tests/models/vit
History
Michael Benayoun 8c7481f35c
ViT and Swin symbolic tracing with torch.fx (#17182) 
* Support tracing for ViT

* Swin support

* Fix copies

* Fix type annotation issue

* Removed unused import
2022-05-12 10:42:27 +02:00
..
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_feature_extraction_vit.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_flax_vit.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_tf_vit.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00
test_modeling_vit.py ViT and Swin symbolic tracing with torch.fx (#17182) 2022-05-12 10:42:27 +02:00