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'☆338Updated 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.☆697Updated 3 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆596Updated 5 years ago
- Compute CNN receptive field size in pytorch in one line☆361Updated last year
- Unofficial PyTorch Reimplementation of RandAugment.☆636Updated 2 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆991Updated 9 months ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆347Updated last year
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆800Updated last year
- A New Optimization Technique for Deep Neural Networks☆537Updated 3 years ago
- PyTorch implementation of AutoAugment.☆159Updated 5 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆337Updated 5 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆170Updated 2 years ago
- Official Pytorch implementation of CutMix regularizer☆1,242Updated 4 years ago
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆328Updated 4 months ago
- CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.☆354Updated 3 years ago
- 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
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆990Updated 2 months ago
- torchsummaryX: Improved visualization tool of torchsummary☆303Updated 3 years ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆417Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆313Updated 5 years ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,043Updated 3 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆764Updated 4 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 3 years ago
- SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)☆128Updated 4 years ago
- Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models☆452Updated 3 years ago
- Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"☆185Updated last year
- Image Test Time Augmentation with PyTorch!☆1,017Updated 2 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆883Updated 5 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆584Updated 2 years ago