pabloppp / pytorch-tools
Useful PyTorch functions and modules that are not implemented in PyTorch by default
☆187Updated 11 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
- Model Scope in PyTorch (include Params, FLOPs, Madds).☆121Updated 5 years ago
- A PyTorch implementation of " EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks."☆311Updated 5 years ago
- Mish Deep Learning Activation Function for PyTorch / FastAI☆161Updated 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
- Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".☆159Updated 4 years ago
- Standardizing weights to accelerate micro-batch training☆547Updated 3 years ago
- Over9000 optimizer☆427Updated 2 years ago
- A simpler version of the self-attention layer from SAGAN, and some image classification results.☆212Updated 5 years ago
- Semi-supervised ImageNet1K models☆243Updated 5 years ago
- Implementing Lambda Networks using Pytorch☆139Updated 4 years ago
- 🛠 Toolbox to extend PyTorch functionalities☆418Updated 10 months ago
- Distributed, mixed-precision training with PyTorch☆89Updated 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
- lookahead optimizer (Lookahead Optimizer: k steps forward, 1 step back) for pytorch☆334Updated 5 years ago
- Utilities for Pytorch☆89Updated 2 years ago
- Implementation of the 😇 Attention layer from the paper, Scaling Local Self-Attention For Parameter Efficient Visual Backbones☆198Updated 4 years ago
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- How to use Cross Replica / Synchronized Batchnorm in Pytorch☆249Updated 5 years ago
- Pytorch implementation of Robust Bi-Tempered Logistic Loss Based on Bregman Divergences☆62Updated 5 years ago
- PyTorch implementation of "Filter Response Normalization Layer: Eliminating Batch Dependence in the Training of Deep Neural Networks"☆86Updated 5 years ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆319Updated last year
- Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)☆473Updated 3 months ago
- ☆191Updated 8 months ago
- Pytorch implementation of the hamburger module from the ICLR 2021 paper "Is Attention Better Than Matrix Decomposition"☆98Updated 4 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆329Updated 2 years ago
- opencv reimplement for transforms in torchvision☆194Updated last year
- Gradient based receptive field estimation for Convolutional Neural Networks☆338Updated 5 years ago
- homura is a library for fast prototyping DL research☆107Updated 2 years ago
- Labels and other data for the paper "Are we done with ImageNet?"☆190Updated 3 years ago