🛠 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