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

118 KiB

Executable File

Raw Blame History

View Raw