PistonY / torch-toolbox
🛠Toolbox to extend PyTorch functionalities
☆416Updated 4 months ago
Related projects: ⓘ
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆331Updated 3 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆627Updated last year
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆687Updated 2 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆585Updated 4 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆469Updated last year
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆781Updated 7 months ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆309Updated 4 years ago
- Compute CNN receptive field size in pytorch in one line☆348Updated 4 months ago
- PyTorch implementation of AutoAugment.☆157Updated 4 years ago
- AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty☆975Updated last month
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆971Updated 3 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆332Updated 2 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆333Updated 5 years ago
- Official Pytorch implementation of CutMix regularizer☆1,214Updated 4 years ago
- CVPR 2020 oral paper: Overcoming Classifier Imbalance for Long-tail Object Detection with Balanced Group Softmax.☆352Updated 2 years ago
- Class-Balanced Loss Based on Effective Number of Samples. CVPR 2019☆599Updated 3 years ago
- Pytorch implementation for "Large-Scale Long-Tailed Recognition in an Open World" (CVPR 2019 ORAL)☆841Updated 2 years ago
- This repository contains code for the paper "Decoupling Representation and Classifier for Long-Tailed Recognition", published at ICLR 202…☆938Updated 2 years ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆411Updated 4 years ago
- Standardizing weights to accelerate micro-batch training☆546Updated 2 years ago
- Code for Noisy Student Training. https://arxiv.org/abs/1911.04252☆750Updated 3 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆579Updated last year
- mixup: Beyond Empirical Risk Minimization☆1,156Updated 2 years ago
- 2.56%, 15.20%, 1.30% on CIFAR10, CIFAR100, and SVHN https://arxiv.org/abs/1708.04552☆542Updated 4 years ago
- The official PyTorch implementation of paper BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition☆657Updated last year
- Image Test Time Augmentation with PyTorch!☆965Updated last year
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆881Updated 4 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆587Updated 2 years ago
- PyTorch Implementation for Deep Metric Learning Pipelines☆575Updated 4 years ago
- Implementation of the mixup training method☆460Updated 6 years ago