Simple package that makes your generator work in background thread
☆281Jun 9, 2022Updated 3 years ago
Alternatives and similar repositories for prefetch_generator
Users that are interested in prefetch_generator are comparing it to the libraries listed below
Sorting:
- Prefetch elements from a Python generator in the background, from a separate process☆34Mar 25, 2021Updated 4 years ago
- My best practice of training large dataset using PyTorch.☆1,104May 9, 2024Updated last year
- high performance image loading and augmenting routines mimicking PIL.Image interface☆318Aug 16, 2021Updated 4 years ago
- An unofficial styleguide and best practices summary for PyTorch☆2,013Dec 28, 2021Updated 4 years ago
- PyTorch layer-by-layer model profiler☆606May 23, 2021Updated 4 years ago
- High-level batteries-included neural network training library for Pytorch☆402Feb 4, 2022Updated 4 years ago
- A plug-in replacement for DataLoader to load Imagenet disk-sequentially in PyTorch.☆239Aug 18, 2021Updated 4 years ago
- Efficient Data Loading Pipeline in Pure Python☆213Aug 19, 2020Updated 5 years ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep lear…☆5,637Updated this week
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Jul 4, 2020Updated 5 years ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,064Mar 2, 2024Updated 2 years ago
- Library for faster pinned CPU <-> GPU transfer in Pytorch☆683Feb 21, 2020Updated 6 years ago
- Debug PyTorch code using PySnooper☆800Apr 28, 2021Updated 4 years ago
- A Toolkit for Training, Tracking, Saving Models and Syncing Results☆62Mar 12, 2020Updated 5 years ago
- An optimizer that trains as fast as Adam and as good as SGD.☆2,911Jul 23, 2023Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- Accelerate training by storing parameters in one contiguous chunk of memory.☆294Oct 29, 2020Updated 5 years ago
- Pytorch implementation of newly added convolution☆584Oct 26, 2019Updated 6 years ago
- implementation of relationNet naive version☆12Dec 4, 2017Updated 8 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- A lightweight library for PyTorch training tools and utilities☆1,720Feb 10, 2026Updated 3 weeks ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- Deep Learning Experiment Management☆641Dec 11, 2025Updated 2 months ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Mar 28, 2020Updated 5 years ago
- All-in-one Toolbox for Computer Vision Research.☆661Mar 10, 2023Updated 2 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆602Dec 27, 2019Updated 6 years ago
- torchbearer: A model fitting library for PyTorch☆641Dec 4, 2023Updated 2 years ago
- A utility library around PyTorch☆244Dec 26, 2022Updated 3 years ago
- Code for training on Imagenet to SOTA results using PyTorch☆13Aug 14, 2023Updated 2 years ago
- Model analyzer in PyTorch☆1,501Mar 19, 2023Updated 2 years ago
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆15,268Jun 25, 2025Updated 8 months ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- Image Test Time Augmentation with PyTorch!☆1,028Jul 28, 2023Updated 2 years ago
- Easy to use Pytorch☆70Mar 21, 2025Updated 11 months ago
- Full implementation of the paper "Rethinking Softmax with Cross-Entropy: Neural Network Classifier as Mutual Information Estimator".☆101Mar 9, 2020Updated 5 years ago
- NeurIPS 2019 Paper Implementation☆12Nov 22, 2022Updated 3 years ago
- Implementations of ideas from recent papers☆391Dec 22, 2020Updated 5 years ago
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,749Updated this week