etienne87 / pytorch-stream-dataloader
☆49Updated 2 years ago
Alternatives and similar repositories for pytorch-stream-dataloader:
Users that are interested in pytorch-stream-dataloader are comparing it to the libraries listed below
- 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
- Implementation of OmniNet, Omnidirectional Representations from Transformers, in Pytorch☆57Updated 4 years ago
- Code for the paper PermuteFormer☆42Updated 3 years ago
- ☆109Updated 4 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 3 years ago
- ☆47Updated 4 years ago
- ☆62Updated 4 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆128Updated 5 months ago
- Distributed, mixed-precision training with PyTorch☆90Updated 4 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated last year
- Pytorch implementation of CVPR2021 paper: SuperMix: Supervising the Mixing Data Augmentation☆92Updated 3 years ago
- ☆38Updated 2 years ago
- AlphaNet Improved Training of Supernet with Alpha-Divergence☆98Updated 3 years ago
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 3 years ago
- ☆25Updated 4 years ago
- Implementation of Memformer, a Memory-augmented Transformer, in Pytorch☆115Updated 4 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- custom cuda kernel for {2, 3}d relative attention with pytorch wrapper☆43Updated 4 years ago
- [ICLR 2022] "As-ViT: Auto-scaling Vision Transformers without Training" by Wuyang Chen, Wei Huang, Xianzhi Du, Xiaodan Song, Zhangyang Wa…☆76Updated 3 years ago
- An implementation of 2021 paper by Geoffrey Hinton: "How to represent part-whole hierarchies in a neural network" in Pytorch.☆58Updated 4 years ago
- ☆29Updated 3 years ago
- PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations☆16Updated 5 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- MTAdam: Automatic Balancing of Multiple Training Loss Terms☆36Updated 4 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
- Demystify RAM Usage in Multi-Process Data Loaders☆193Updated 2 years ago
- Official implementation of "UNAS: Differentiable Architecture Search Meets Reinforcement Learning", CVPR 2020 Oral☆60Updated last year