🛠 Toolbox to extend PyTorch functionalities
☆421May 13, 2024Updated last year
Alternatives and similar repositories for torch-toolbox
Users that are interested in torch-toolbox are comparing it to the libraries listed below
Sorting:
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,567Oct 9, 2025Updated 4 months ago
- Image Test Time Augmentation with PyTorch!☆1,028Jul 28, 2023Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,260Oct 17, 2024Updated last year
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆478Dec 10, 2024Updated last year
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆702Dec 24, 2021Updated 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,612Jun 16, 2021Updated 4 years ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Jan 1, 2024Updated 2 years ago
- Slicing a PyTorch Tensor Into Parallel Shards☆300Jun 7, 2025Updated 8 months ago
- This repo parallelizes mAP_evaluation using python's multiprocessing module.☆18Apr 14, 2022Updated 3 years ago
- ☆39Aug 26, 2020Updated 5 years ago
- A PyTorch implementation of EfficientNet☆8,221Apr 8, 2022Updated 3 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆339Mar 26, 2021Updated 4 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Jul 4, 2020Updated 5 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,168Mar 20, 2024Updated last year
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,497May 16, 2023Updated 2 years ago
- ResNeSt: Split-Attention Networks☆3,267Dec 9, 2022Updated 3 years ago
- Over9000 optimizer☆424Nov 22, 2022Updated 3 years ago
- homura is a library for fast prototyping DL research☆106Jun 10, 2022Updated 3 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988May 30, 2025Updated 9 months ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,325Jan 5, 2026Updated last month
- Hands on Imagenet training. Unofficial ModelZoo project on Pytorch. MobileNetV3 Top1 75.64🌟 GhostNet1.3x 75.78🌟☆49Nov 30, 2020Updated 5 years ago
- Python library for various computer vision problems with a focus on easy usage.☆18Jan 25, 2021Updated 5 years ago
- Revisiting Parameter Sharing for Automatic Neural Channel Number Search, NeurIPS 2020☆22Nov 15, 2020Updated 5 years ago
- The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition☆668Nov 22, 2022Updated 3 years ago
- A real-time GNN-based method. Understanding Image Retrieval Re-Ranking: A Graph Neural Network Perspective☆77Mar 3, 2021Updated 5 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,458Feb 10, 2023Updated 3 years ago
- A New Optimization Technique for Deep Neural Networks☆539Jan 13, 2022Updated 4 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆294Oct 29, 2020Updated 5 years ago
- Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models☆452Jan 30, 2022Updated 4 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Jul 31, 2021Updated 4 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Jul 29, 2020Updated 5 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- face recognition training project(pytorch)☆484Nov 12, 2021Updated 4 years ago
- Tool box for PyTorch☆187Apr 8, 2025Updated 10 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆223Mar 24, 2025Updated 11 months ago
- Sandbox for training deep learning networks☆3,020Sep 6, 2024Updated last year
- My best practice of training large dataset using PyTorch.☆1,104May 9, 2024Updated last year
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆806Feb 18, 2024Updated 2 years ago