PistonY / torch-toolboxLinks
🛠 Toolbox to extend PyTorch functionalities
☆420Updated 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'☆340Updated 4 years ago
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆701Updated 4 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆478Updated last year
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆594Updated 5 years ago
- Compute CNN receptive field size in pytorch in one line☆367Updated last year
- A New Optimization Technique for Deep Neural Networks☆540Updated 3 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆805Updated last year
- PyTorch implementation of AutoAugment.☆161Updated 5 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆338Updated 6 years ago
- torchsummaryX: Improved visualization tool of torchsummary☆302Updated 3 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆638Updated 2 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆992Updated last year
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆421Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆315Updated 6 years ago
- PyTorch implementations of recent Computer Vision tricks (ReXNet, RepVGG, Unet3p, YOLOv4, CIoU loss, AdaBelief, PolyLoss, MobileOne). Oth…☆328Updated 3 weeks ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆350Updated last year
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆170Updated 3 years ago
- CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.☆356Updated 3 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆884Updated 5 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆338Updated 4 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
- PyTorch implementation of Octave Convolution with pre-trained Oct-ResNet and Oct-MobileNet models☆293Updated 5 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆988Updated 7 months ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆765Updated 4 years ago
- This repository reproduces the results of the paper: "Fixing the train-test resolution discrepancy" https://arxiv.org/abs/1906.06423☆1,044Updated 4 years ago
- Official Pytorch implementation of CutMix regularizer☆1,249Updated 5 years ago
- Implementation of Dynamic ReLU on Pytorch☆205Updated 5 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆585Updated 2 years ago
- Visualization toolkit for neural networks in PyTorch! Demo -->☆743Updated 2 years ago
- Standardizing weights to accelerate micro-batch training☆549Updated 3 years ago