Commit Graph

9 Commits (ff21fdb41d310368ede8fa2ed293901a6b03da2c)

Author SHA1 Message Date
Ross Wightman 5f47518f27 Fix pit implementation to be clsoer to deit/levit re distillation head handling 3 years ago
Ross Wightman 372ad5fa0d Significant model refactor and additions: 3 years ago
Ross Wightman 5f81d4de23 Move DeiT to own file, vit getting crowded. Working towards fixing , make pooling interface for transformers and mlp closer to convnets. Still working through some details... 3 years ago
Ross Wightman abc9ba2544 Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks. 3 years ago
Ross Wightman 8880f696b6 Refactoring, cleanup, improved test coverage. 4 years ago
Ross Wightman bfc72f75d3 Expand scope of testing for non-std vision transformer / mlp models. Some related cleanup and create fn cleanup for all vision transformer and mlp models. More CoaT weights. 4 years ago
Ross Wightman f606c45c38 Add Swin Transformer models from https://github.com/microsoft/Swin-Transformer 4 years ago
Ross Wightman 7953e5d11a Fix pos_embed scaling for ViT and num_classes != 1000 for pretrained distilled deit and pit models. Fix and fix 4 years ago
Ross Wightman 0dfc5a66bb Add PiT model from https://github.com/naver-ai/pit 4 years ago