transformers
8c7481f3 - ViT and Swin symbolic tracing with torch.fx (#17182)

Commit
3 years ago
ViT and Swin symbolic tracing with torch.fx (#17182) * Support tracing for ViT * Swin support * Fix copies * Fix type annotation issue * Removed unused import
Parents
Loading