PistonY / torch-toolbox
🛠Toolbox to extend PyTorch functionalities
☆417Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for torch-toolbox
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆332Updated 3 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆588Updated 4 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆785Updated 9 months ago
- Unofficial PyTorch Reimplementation of RandAugment.☆628Updated last year
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆693Updated 2 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆471Updated this week
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆977Updated last month
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆309Updated 4 years ago
- Official Pytorch implementation of CutMix regularizer☆1,221Updated 4 years ago
- This repository contains code for the paper "Decoupling Representation and Classifier for Long-Tailed Recognition", published at ICLR 202…☆946Updated 3 years ago
- CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.☆351Updated 2 years ago
- Compute CNN receptive field size in pytorch in one line☆349Updated 6 months ago
- PyTorch implementation of AutoAugment.☆157Updated 4 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆580Updated last year
- Standardizing weights to accelerate micro-batch training☆546Updated 2 years ago
- torchsummaryX: Improved visualization tool of torchsummary☆301Updated 2 years ago
- My best practice of training large dataset using PyTorch.☆1,086Updated 6 months ago
- Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)☆849Updated 2 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆980Updated 3 months ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆414Updated 4 years ago
- mixup: Beyond Empirical Risk Minimization☆1,168Updated 3 years ago
- a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.☆167Updated 2 years ago
- A New Optimization Technique for Deep Neural Networks☆533Updated 2 years ago
- Image Test Time Augmentation with PyTorch!☆978Updated last year
- Official Pytorch implementation of ReXNet (Rank eXpansion Network) with pretrained models☆453Updated 2 years ago
- The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition☆661Updated last year
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆753Updated 3 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- PyTorch Implementation for Deep Metric Learning Pipelines☆572Updated 4 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆880Updated 4 years ago