Alternatives and similar repositories for pytorch_warmup

Users that are interested in pytorch_warmup are comparing it to the libraries listed below

• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆444Updated last year
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆468Updated 2 years ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆992Updated last year
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,081Updated 3 years ago
• AdeelH / pytorch-multi-class-focal-loss
  An (unofficial) implementation of Focal Loss, as described in the RetinaNet paper, generalized to the multi-class case.
  ☆239Updated 2 years ago
• Fangyh09 / pytorch-receptive-field
  Compute CNN receptive field size in pytorch in one line
  ☆367Updated last year
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆261Updated 5 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,056Updated 7 months ago
• lucidrains / transformer-in-transformer
  Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…
  ☆310Updated 4 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆824Updated 3 years ago
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆485Updated 4 years ago
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆421Updated last year
• imbue-ai / self_supervised
  A Pytorch-Lightning implementation of self-supervised algorithms
  ☆545Updated 3 years ago
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆350Updated 2 years ago
• SHI-Labs / Compact-Transformers
  Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)
  ☆537Updated last year
• yangkky / distributed_tutorial
  ☆262Updated 6 years ago
• lucidrains / axial-attention
  Implementation of Axial attention - attending to multi-dimensional data efficiently
  ☆397Updated 4 years ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆340Updated last year
• facebookresearch / convit
  Code for the Convolutional Vision Transformer (ConViT)
  ☆472Updated 4 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆305Updated 2 years ago
• vahidk / tfrecord
  Standalone TFRecord reader/writer with PyTorch data loaders
  ☆900Updated 8 months ago
• cmsflash / efficient-attention
  An implementation of the efficient attention module.
  ☆328Updated 5 years ago
• SwinTransformer / Transformer-SSL
  This is an official implementation for "Self-Supervised Learning with Swin Transformers".
  ☆665Updated 4 years ago
• frgfm / torch-scan
  Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)
  ☆224Updated 10 months ago
• rishikksh20 / FNet-pytorch
  Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms
  ☆261Updated 4 years ago
• google-research / noisystudent
  Code for Noisy Student Training. https://arxiv.org/abs/1911.04252
  ☆763Updated 4 years ago
• clcarwin / focal_loss_pytorch
  A PyTorch Implementation of Focal Loss.
  ☆992Updated 6 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆827Updated last year
• sooftware / pytorch-lr-scheduler
  PyTorch implementation of some learning rate schedulers for deep learning researcher.
  ☆91Updated 3 years ago
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆516Updated 2 years ago