pabloppp / pytorch-tools
Useful PyTorch functions and modules that are not implemented in PyTorch by default
☆187Updated 10 months ago
Alternatives and similar repositories for pytorch-tools:
Users that are interested in pytorch-tools are comparing it to the libraries listed below
- Unofficial PyTorch Implementation of EvoNorm☆121Updated 3 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
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆309Updated 5 years ago
- Mish Deep Learning Activation Function for PyTorch / FastAI☆161Updated 4 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆121Updated 5 years ago
- Implementing Lambda Networks using Pytorch☆138Updated 4 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 3 years ago
- Over9000 optimizer☆427Updated 2 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆419Updated 9 months ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆217Updated this week
- Implements https://arxiv.org/abs/1711.05101 AdamW optimizer, cosine learning rate scheduler and "Cyclical Learning Rates for Training Neu…☆149Updated 5 years ago
- Standardizing weights to accelerate micro-batch training☆547Updated 3 years ago
- opencv reimplement for transforms in torchvision☆194Updated last year
- Accelerate training by storing parameters in one contiguous chunk of memory.☆292Updated 4 years ago
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆472Updated 2 months ago
- A pytorch dataset sampler for always sampling balanced batches.☆111Updated 4 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆344Updated last year
- ☆182Updated 2 years ago
- Feature extraction made simple with torchextractor☆100Updated 3 years ago
- homura is a library for fast prototyping DL research☆107Updated 2 years ago
- Tool box for PyTorch☆187Updated 11 months ago
- Utilities for Pytorch☆89Updated 2 years ago
- Gradient based receptive field estimation for Convolutional Neural Networks☆338Updated 5 years ago
- Deal with bad samples in your dataset dynamically, use Transforms as Filters, and more!☆377Updated 2 years ago
- Implementations of ideas from recent papers☆391Updated 4 years ago
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆198Updated 3 years ago
- A simpler version of the self-attention layer from SAGAN, and some image classification results.☆213Updated 5 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆328Updated 2 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 6 years ago