nmhkahn / torchsummaryXLinks
torchsummaryX: Improved visualization tool of torchsummary
☆303Updated 3 years ago
Alternatives and similar repositories for torchsummaryX
Users that are interested in torchsummaryX are comparing it to the libraries listed below
Sorting:
- PyTorch implementation of Octave Convolution with pre-trained Oct-ResNet and Oct-MobileNet models☆294Updated 5 years ago
- Pytorch implementation of newly added convolution☆586Updated 5 years ago
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆596Updated 4 years ago
- A New Optimization Technique for Deep Neural Networks☆537Updated 3 years ago
- Code for paper☆560Updated 3 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆989Updated 9 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆421Updated last year
- Compute CNN receptive field size in pytorch in one line☆360Updated last year
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆337Updated 5 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆654Updated 3 years ago
- Implementing Stand-Alone Self-Attention in Vision Models using Pytorch☆455Updated 5 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆882Updated 5 years ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆845Updated 2 years ago
- Implementation of the mixup training method☆466Updated 7 years ago
- PyTorch implementation of AutoAugment.☆159Updated 5 years ago
- Implementation of Dynamic ReLU on Pytorch☆204Updated 5 years ago
- My best practice of training large dataset using PyTorch.☆1,099Updated last year
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆866Updated 4 years ago
- Standardizing weights to accelerate micro-batch training☆549Updated 3 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆475Updated 7 months ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆418Updated 5 years ago
- Pyramidal Convolution: Rethinking Convolutional Neural Networks for Visual Recognition (https://arxiv.org/pdf/2006.11538.pdf)☆333Updated 4 years ago
- MEAL V2: Boosting Vanilla ResNet-50 to 80%+ Top-1 Accuracy on ImageNet without Tricks. In NeurIPS 2020 workshop.☆696Updated 3 years ago
- PyTorch Implementation of Deformable Convolution☆288Updated 7 years ago
- Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu:https://zhuanlan.zhihu.com/p/59690223☆610Updated 6 years ago
- Exploring Self-attention for Image Recognition, CVPR2020.☆752Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆313Updated 5 years ago
- Official implementation of 'FMix: Enhancing Mixed Sample Data Augmentation'☆337Updated 4 years ago
- Source code for "On the Relationship between Self-Attention and Convolutional Layers"☆1,103Updated 2 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆598Updated 5 years ago