tmbdev-archive / webdataset-examples
Examples for the WebDataset PyTorch Dataset Library
☆47Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for webdataset-examples
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆73Updated 11 months ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆73Updated 2 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)☆20Updated last year
- ☆49Updated last year
- Fast and simple stream processing of files in tar files, useful for deep learning, big data, and many other applications.☆123Updated 11 months ago
- ☆25Updated 3 years ago
- [ICLR 2022] "As-ViT: Auto-scaling Vision Transformers without Training" by Wuyang Chen, Wei Huang, Xianzhi Du, Xiaodan Song, Zhangyang Wa…☆76Updated 2 years ago
- ☆61Updated 4 years ago
- This is a offical PyTorch/GPU implementation of SupMAE.☆77Updated 2 years ago
- Demystify RAM Usage in Multi-Process Data Loaders☆183Updated last year
- This is the official PyTorch implementation for "Mesa: A Memory-saving Training Framework for Transformers".☆118Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".☆83Updated 9 months ago
- Implementation of RQ Transformer, proposed in the paper "Autoregressive Image Generation using Residual Quantization"☆95Updated 2 years ago
- A small repository demonstrating the use of Webdataset and Imagenet☆14Updated 11 months ago
- PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)☆138Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- A non-JIT version implementation / replication of CLIP of OpenAI in pytorch☆34Updated 3 years ago
- [NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "EcoFormer: Energy-Saving Attention with Linear Complexity"☆66Updated 2 years ago
- ViT trained on COYO-Labeled-300M dataset☆29Updated 2 years ago
- This is a PyTorch implementation of the paperViP A Differentially Private Foundation Model for Computer Vision☆37Updated last year
- Simple python template☆40Updated 6 months ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated last year
- PyTorch implementation of IRMAE https//arxiv.org/abs/2010.00679☆45Updated 2 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Release of ImageNet-Captions☆45Updated last year
- Patching open-vocabulary models by interpolating weights☆90Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆207Updated last year