fadel / pytorch_ema

Related projects: ⓘ

• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆387Updated 4 months ago
• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆433Updated last year
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your pytorch model
  ☆473Updated 3 weeks ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆332Updated 4 years ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆806Updated 2 years ago
• AndrewAtanov / simclr-pytorch
  PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results
  ☆187Updated 4 months ago
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆322Updated 7 months ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆971Updated 3 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆244Updated 8 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆672Updated 4 months ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆987Updated this week
• sthalles / PyTorch-BYOL
  PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
  ☆476Updated 2 years ago
• microsoft / esvit
  EsViT: Efficient self-supervised Vision Transformers
  ☆407Updated last year
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,057Updated last year
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,078Updated last year
• imbue-ai / self_supervised
  A Pytorch-Lightning implementation of self-supervised algorithms
  ☆530Updated 2 years ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆251Updated 3 years ago
• facebookresearch / msn
  Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
  ☆447Updated 2 years ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆207Updated 3 years ago
• nocotan / pytorch-lightning-gans
  Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.
  ☆164Updated 2 years ago
• NVIDIA / transformer-ls
  Official PyTorch Implementation of Long-Short Transformer (NeurIPS 2021).
  ☆221Updated 2 years ago
• google-research / sam
  ☆552Updated last month
• lukemelas / do-you-even-need-attention
  Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)
  ☆480Updated 3 years ago
• NVlabs / NVAE
  The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
  ☆1,012Updated last year
• sail-sg / Adan
  Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
  ☆746Updated 2 months ago
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆343Updated 8 months ago
• lucidrains / slot-attention
  Implementation of Slot Attention from GoogleAI
  ☆378Updated 3 weeks ago
• bytedance / ibot
  iBOT : Image BERT Pre-Training with Online Tokenizer (ICLR 2022)
  ☆665Updated 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!)
  ☆492Updated last year
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆416Updated 4 months ago