CEA-LIST / RPCDataloader
A variant of the PyTorch Dataloader using remote workers.
☆18Updated 2 years ago
Alternatives and similar repositories for RPCDataloader:
Users that are interested in RPCDataloader are comparing it to the libraries listed below
- Demystify RAM Usage in Multi-Process Data Loaders☆191Updated last year
- Library to perform efficient vision data augmentations for CPU/GPU per-sample/batched data.☆24Updated last month
- Approximate nearest neighbor search with product quantization on GPU in pytorch and cuda☆221Updated last year
- Data-centric AI building blocks for computer vision applications☆47Updated this week
- Examples for the WebDataset PyTorch Dataset Library☆47Updated 3 years ago
- gpu tester detects broken and slow gpus in a cluster☆68Updated 2 years ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆229Updated 2 months ago
- A python library that provides common I/O interface across different storage backends.☆143Updated last month
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆77Updated 2 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆212Updated 2 years ago
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆252Updated 2 years ago
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆139Updated 3 years ago
- The PASS dataset: pretrained models and how to get the data☆263Updated 2 years ago
- A compilation of network architectures for vision and others without usage of self-attention mechanism☆77Updated 2 years ago
- ☆62Updated 4 years ago
- ☆50Updated last year
- Customized matrix multiplication kernels☆54Updated 3 years ago
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- MONeT framework for reducing memory consumption of DNN training☆173Updated 3 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆125Updated 3 years ago
- Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"☆82Updated 2 weeks ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- Code and models for the paper "The effectiveness of MAE pre-pretraining for billion-scale pretraining" https://arxiv.org/abs/2303.13496☆88Updated 8 months ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆212Updated 2 years ago
- ☆84Updated this week
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 3 years ago
- ☆29Updated 6 months ago
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆42Updated 3 years ago
- Pytorch implementation of LOST unsupervised object discovery method☆242Updated last year