π Toolbox to extend PyTorch functionalities
β420May 13, 2024Updated 2 years ago
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,573Oct 9, 2025Updated 7 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,261Oct 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 4 years ago
- Proton VPN Special Offer - Get 70% off β’ AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 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 4 years ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.β1,607Jun 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 5 years ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillowβ1,496May 16, 2023Updated 3 years ago
- Over9000 optimizerβ424Nov 22, 2022Updated 3 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessingβ883Jul 4, 2020Updated 5 years ago
- GPUs on demand by Runpod - Special Offer Available β’ AdRun AI, ML, and HPC workloads on powerful cloud GPUsβwithout limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- ResNeSt: Split-Attention Networksβ3,263Dec 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 11 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β991Mar 12, 2026Updated 2 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.β9,109Apr 22, 2022Updated 4 years ago
- Virtual machines for every use case on DigitalOcean β’ AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- 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,474Feb 10, 2023Updated 3 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --β¦β36,809May 8, 2026Updated 2 weeks ago
- β147Feb 21, 2021Updated 5 years ago
- A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.β2,323Apr 6, 2026Updated last month
- 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,083Jul 8, 2024Updated last year
- The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognitionβ667Nov 22, 2022Updated 3 years ago
- The First Place Solution of Kaggle iMaterialist (Fashion) 2019 at FGVC6β484Jan 12, 2020Updated 6 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer β’ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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,322Aug 17, 2025Updated 9 months ago
- Some tricks of pytorch...β1,191Jun 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.)β223Mar 24, 2025Updated last year
- Tool box for PyTorchβ186Apr 8, 2025Updated last year