PistonY / torch-toolboxLinks
🛠 Toolbox to extend PyTorch functionalities
☆421Updated 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:
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆337Updated 4 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆475Updated 7 months ago
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆696Updated 3 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆596Updated 4 years ago
- Compute CNN receptive field size in pytorch in one line☆360Updated last year
- A New Optimization Technique for Deep Neural Networks☆537Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆799Updated last year
- PyTorch implementation of AutoAugment.☆159Updated 5 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆989Updated 9 months ago
- Unofficial PyTorch Reimplementation of RandAugment.☆637Updated 2 years ago
- CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.☆354Updated 3 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆170Updated 2 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆337Updated 5 years ago
- torchsummaryX: Improved visualization tool of torchsummary☆303Updated 3 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆347Updated last year
- Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia …☆197Updated 5 years ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆418Updated 5 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆990Updated last month
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆329Updated 3 months ago
- Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"☆185Updated 11 months ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆313Updated 5 years ago
- Image Test Time Augmentation with PyTorch!☆1,016Updated last year
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,042Updated 3 years ago
- Official Pytorch implementation of CutMix regularizer☆1,240Updated 4 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆761Updated 4 years ago
- ☆261Updated 5 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 3 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- Search for RegNet using PyTorch☆316Updated 5 years ago
- SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)☆128Updated 4 years ago