A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.
☆141Dec 21, 2021Updated 4 years ago
Alternatives and similar repositories for Image-Local-Attention
Users that are interested in Image-Local-Attention are comparing it to the libraries listed below
Sorting:
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Dec 4, 2021Updated 4 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆517Sep 28, 2023Updated 2 years ago
- Contrastive Language-Image Pretraining☆144Sep 6, 2022Updated 3 years ago
- (CVPR 2022) Official PyTorch implementation of "Reflection and Rotation Detection via Equivariant Learning."☆25Jul 26, 2022Updated 3 years ago
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆563Jul 13, 2023Updated 2 years ago
- Texture mapping with variational auto-encoders☆40Oct 1, 2021Updated 4 years ago
- [ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"☆291Apr 25, 2022Updated 3 years ago
- [NeurIPS 2021] [T-PAMI] DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification☆650Jul 11, 2023Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Aug 28, 2021Updated 4 years ago
- Code release for SLIP Self-supervision meets Language-Image Pre-training☆787Feb 9, 2023Updated 3 years ago
- [NeurIPS 2021] ORL: Unsupervised Object-Level Representation Learning from Scene Images☆58Dec 6, 2021Updated 4 years ago
- Contrastive Learning of Image Representations with Cross-Video Cycle-Consistency☆17Dec 2, 2021Updated 4 years ago
- Anytime Dense Prediction with Confidence Adaptivity (ICLR 2022)☆51Aug 23, 2024Updated last year
- ImageNet-12k subset of ImageNet-21k (fall11)☆21Jun 13, 2023Updated 2 years ago
- [ECCV2022] Revisiting the Critical Factors of Augmentation-Invariant Representation Learning☆12Aug 3, 2022Updated 3 years ago
- ☆80Jan 21, 2022Updated 4 years ago
- Udacity CS344 Introduction to Parallell Programming (https://classroom.udacity.com/courses/cs344), with assignments/materials updated to …☆45Sep 16, 2021Updated 4 years ago
- Adam with minor modifications which give significant improvement☆19Aug 20, 2021Updated 4 years ago
- A PyTorch-based Semi-Supervised Learning (SSL) Codebase for Pixel-wise (Pixel) Vision Tasks [ECCV 2020]☆290Apr 6, 2021Updated 4 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆58Jan 13, 2021Updated 5 years ago
- [ICLR 2023 Spotlight] GPViT: A High Resolution Non-Hierarchical Vision Transformer with Group Propagation☆103May 26, 2023Updated 2 years ago
- Official PyTorch implementation of Fully Attentional Networks☆482Mar 31, 2023Updated 2 years ago
- JAX implementation ViT-VQGAN☆82Sep 21, 2022Updated 3 years ago
- Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"☆345Oct 3, 2022Updated 3 years ago
- Fire Together Wire Together: A Dynamic Pruning Approach with Self-Supervised Mask Prediction☆11May 25, 2022Updated 3 years ago
- Code for the ECCV 2022 paper "Unleashing Transformers"☆185Apr 17, 2023Updated 2 years ago
- Fast, differentiable sorting and ranking in PyTorch☆850Jun 9, 2025Updated 8 months ago
- Object detection on multiple datasets with an automatically learned unified label space.☆517Mar 8, 2024Updated last year
- Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).☆228Apr 18, 2022Updated 3 years ago
- Official code Cross-Covariance Image Transformer (XCiT)☆674Sep 28, 2021Updated 4 years ago
- Official Codes for "Uniform Masking: Enabling MAE Pre-training for Pyramid-based Vision Transformers with Locality"☆245Dec 3, 2022Updated 3 years ago
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,829Jan 18, 2026Updated last month
- Memory-efficient transformer. Work in progress.☆19Sep 17, 2022Updated 3 years ago
- BigDetection: A Large-scale Benchmark for Improved Object Detector Pre-training☆401Oct 23, 2024Updated last year
- Simple implementation of dense visual odometry☆46Oct 16, 2018Updated 7 years ago
- Supervision Exists Everywhere: A Data Efficient Contrastive Language-Image Pre-training Paradigm☆675Sep 19, 2022Updated 3 years ago
- This is the official PyTorch implementation for "Mesa: A Memory-saving Training Framework for Transformers".☆120Dec 12, 2021Updated 4 years ago
- [NeurIPS 2021] Revitalizing CNN Attentions via Transformers in Self-Supervised Visual Representation Learning☆110Nov 9, 2021Updated 4 years ago
- ☆73Aug 26, 2020Updated 5 years ago