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