nmhkahn / torchsummaryX
torchsummaryX: Improved visualization tool of torchsummary
☆304Updated 3 years ago
Alternatives and similar repositories for torchsummaryX:
Users that are interested in torchsummaryX are comparing it to the libraries listed below
- Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.☆593Updated 4 years ago
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆988Updated 5 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 10 months ago
- PyTorch implementation of Octave Convolution with pre-trained Oct-ResNet and Oct-MobileNet models☆293Updated 4 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- Pytorch implementation of newly added convolution☆587Updated 5 years ago
- Implementation of the mixup training method☆466Updated 6 years ago
- A New Optimization Technique for Deep Neural Networks☆535Updated 3 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆649Updated 3 years ago
- PyTorch DataLoaders implemented with DALI for accelerating image preprocessing☆880Updated 4 years ago
- My best practice of training large dataset using PyTorch.☆1,092Updated 10 months ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆310Updated 5 years ago
- Implementation of Dynamic ReLU on Pytorch☆204Updated 4 years ago
- A PyTorch Implementation of Focal Loss.☆979Updated 5 years ago
- Compute CNN receptive field size in pytorch in one line☆359Updated 10 months ago
- Implementing Stand-Alone Self-Attention in Vision Models using Pytorch☆454Updated 5 years ago
- Exploring Self-attention for Image Recognition, CVPR2020.☆752Updated 4 years ago
- Pytorch implementation of the paper "Class-Balanced Loss Based on Effective Number of Samples"☆792Updated last year
- PyTorch Implementation of Deformable Convolution☆287Updated 7 years ago
- PyTorch implementation of AutoAugment.☆159Updated 4 years ago
- Official PyTorch implementation of "A Comprehensive Overhaul of Feature Distillation" (ICCV 2019)☆417Updated 4 years ago
- Unofficial PyTorch Reimplementation of RandAugment.☆633Updated 2 years ago
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆596Updated 5 years ago
- Model analyzer in PyTorch☆1,477Updated 2 years ago
- Papers for normalization techniques, released codes collections.☆227Updated 4 years ago
- PyTorch Implementation of Focal Loss and Lovasz-Softmax Loss☆333Updated 3 years ago
- Pyramidal Convolution: Rethinking Convolutional Neural Networks for Visual Recognition (https://arxiv.org/pdf/2006.11538.pdf)☆331Updated 4 years ago
- Standardizing weights to accelerate micro-batch training☆547Updated 3 years ago
- try to implement the CVPR 2019 paper "Selective Kernel Networks" by PyTorch☆372Updated 5 years ago
- Code for paper☆560Updated 3 years ago