🛠 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,568Oct 9, 2025Updated 5 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,262Oct 17, 2024Updated last year
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆478Dec 10, 2024Updated last year
- This repo parallelizes mAP_evaluation using python's multiprocessing module.☆18Apr 14, 2022Updated 3 years ago
- Hands on Imagenet training. Unofficial ModelZoo project on Pytorch. MobileNetV3 Top1 75.64🌟 GhostNet1.3x 75.78🌟☆49Nov 30, 2020Updated 5 years ago
- A PyTorch implementation of EfficientNet☆8,219Apr 8, 2022Updated 3 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,610Jun 16, 2021Updated 4 years ago
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆701Dec 24, 2021Updated 4 years ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Jan 1, 2024Updated 2 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆339Mar 26, 2021Updated 4 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,494May 16, 2023Updated 2 years ago
- Over9000 optimizer☆424Nov 22, 2022Updated 3 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Jul 4, 2020Updated 5 years ago
- ResNeSt: Split-Attention Networks☆3,264Dec 9, 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
- Slicing a PyTorch Tensor Into Parallel Shards☆300Jun 7, 2025Updated 9 months ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,167Mar 22, 2024Updated 2 years ago
- Codebase for Image Classification Research, written in PyTorch.☆2,166Mar 20, 2024Updated 2 years ago
- ☆39Aug 26, 2020Updated 5 years ago
- Python library for various computer vision problems with a focus on easy usage.☆18Jan 25, 2021Updated 5 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988Mar 12, 2026Updated last week
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,109Apr 22, 2022Updated 3 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,464Feb 10, 2023Updated 3 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,538Updated this week
- ☆147Feb 21, 2021Updated 5 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.☆2,324Jan 5, 2026Updated 2 months ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Jul 29, 2020Updated 5 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,080Jul 8, 2024Updated last year
- The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition☆669Nov 22, 2022Updated 3 years ago
- The First Place Solution of Kaggle iMaterialist (Fashion) 2019 at FGVC6☆485Jan 12, 2020Updated 6 years ago
- face recognition training project(pytorch)☆485Nov 12, 2021Updated 4 years ago
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,311Aug 17, 2025Updated 7 months ago
- Some tricks of pytorch...☆1,193Jun 20, 2024Updated last year
- Self-distillation with Batch Knowledge Ensembling Improves ImageNet Classification☆82Jun 9, 2021Updated 4 years ago
- Official Pytorch implementation of CutMix regularizer☆1,246Sep 16, 2020Updated 5 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆222Mar 24, 2025Updated last year
- Tool box for PyTorch☆187Apr 8, 2025Updated 11 months ago