pabloppp / pytorch-toolsLinks
Useful PyTorch functions and modules that are not implemented in PyTorch by default
☆188Updated last year
Alternatives and similar repositories for pytorch-tools
Users that are interested in pytorch-tools are comparing it to the libraries listed below
Sorting:
- Implementing Lambda Networks using Pytorch☆139Updated 4 years ago
- Unofficial PyTorch Implementation of EvoNorm☆122Updated 3 years ago
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 4 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
- Mish Deep Learning Activation Function for PyTorch / FastAI☆161Updated 5 years ago
- A simpler version of the self-attention layer from SAGAN, and some image classification results.☆212Updated 5 years ago
- Implements https://arxiv.org/abs/1711.05101 AdamW optimizer, cosine learning rate scheduler and "Cyclical Learning Rates for Training Neu…☆149Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆313Updated 5 years ago
- opencv reimplement for transforms in torchvision☆194Updated last year
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- Over9000 optimizer☆426Updated 2 years ago
- Tool box for PyTorch☆188Updated 2 months ago
- 🛠 Toolbox to extend PyTorch functionalities☆421Updated last year
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 3 years ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆99Updated 4 years ago
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆199Updated 4 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆218Updated 3 months ago
- Semi-supervised ImageNet1K models☆244Updated 5 years ago
- Feature extraction made simple with torchextractor☆101Updated 4 years ago
- Experimental ground for optimizing memory of pytorch models☆366Updated 7 years ago
- ☆182Updated 2 years ago
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆122Updated 6 years ago
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆337Updated 5 years ago
- Scaling and Benchmarking Self-Supervised Visual Representation Learning☆586Updated 3 years ago
- Standardizing weights to accelerate micro-batch training☆549Updated 3 years ago
- Data Reading Blocks for Python☆105Updated 4 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/☆348Updated last year
- Source code for the paper "Divide and Conquer the Embedding Space for Metric Learning", CVPR 2019☆265Updated 5 years ago
- Implementation and experiments for AdamW on Pytorch☆94Updated 5 years ago