lucidrains / mlp-mixer-pytorch
An All-MLP solution for Vision, from Google AI
β1,003Updated 2 months ago
Related projects β
Alternatives and complementary repositories for mlp-mixer-pytorch
- (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"β806Updated 2 years ago
- Implementation of ConvMixer for "Patches Are All You Need? π€·"β1,062Updated 2 years ago
- PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057β1,218Updated 2 years ago
- Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)β500Updated 2 weeks ago
- PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)β1,295Updated 5 months ago
- PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566β1,162Updated last year
- ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetβ1,149Updated last year
- This is an official implementation for "Self-Supervised Learning with Swin Transformers".β627Updated 3 years ago
- Code for the Convolutional Vision Transformer (ConViT)β462Updated 3 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Visionβ209Updated 3 years ago
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorchβ1,764Updated 4 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)β1,770Updated 9 months ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representationsβ2,265Updated 8 months ago
- A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"β369Updated 3 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.β766Updated 5 months ago
- Transformer based on a variant of attention that is linear complexity in respect to sequence lengthβ698Updated 6 months ago
- This is an official implementation for "SimMIM: A Simple Framework for Masked Image Modeling".β927Updated 2 years ago
- Implementation of the Swin Transformer in PyTorch.β794Updated 3 years ago
- Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)β1,941Updated 2 years ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classificationβ445Updated last year
- Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.β1,182Updated 3 years ago
- Learning Rate Warmup in PyTorchβ392Updated this week
- Pytorch version of Vision Transformer (ViT) with pretrained models. This is part of CASL (https://casl-project.github.io/) and ASYML projβ¦β342Updated 3 years ago
- Recent Transformer-based CV and related works.β1,323Updated last year
- A coding-free framework built on PyTorch for reproducible deep learning studies. π25 knowledge distillation methods presented at CVPR, Iβ¦β1,392Updated last month
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022β1,054Updated 6 months ago
- Vision Transformer (ViT) in PyTorchβ794Updated 2 years ago
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorcβ¦β300Updated 2 years ago
- PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learningβ479Updated 2 years ago
- Implementation of Perceiver, General Perception with Iterative Attention, in Pytorchβ1,097Updated last year