Alternatives and similar repositories for pytorch_ema:

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

• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆450Updated last year
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆559Updated 2 months ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆403Updated 2 weeks ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆341Updated 4 years ago
• microsoft / esvit
  EsViT: Efficient self-supervised Vision Transformers
  ☆411Updated last year
• lucidrains / slot-attention
  Implementation of Slot Attention from GoogleAI
  ☆410Updated 6 months ago
• imbue-ai / self_supervised
  A Pytorch-Lightning implementation of self-supervised algorithms
  ☆537Updated 2 years ago
• sthalles / PyTorch-BYOL
  PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
  ☆483Updated 2 years ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆986Updated 4 months ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆809Updated 2 years ago
• assafshocher / ResizeRight
  The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).
  ☆553Updated last year
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆252Updated 4 years ago
• facebookresearch / msn
  Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
  ☆455Updated 2 years ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆738Updated 9 months ago
• AndrewAtanov / simclr-pytorch
  PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results
  ☆199Updated 9 months ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,064Updated 2 years ago
• cybertronai / pytorch-lamb
  Implementation of https://arxiv.org/abs/1904.00962
  ☆371Updated 4 years ago
• wgrathwohl / JEM
  Project site for "Your Classifier is Secretly an Energy-Based Model and You Should Treat it Like One"
  ☆420Updated 2 years ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,013Updated 5 months ago
• rishikksh20 / MLP-Mixer-pytorch
  Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision
  ☆217Updated 3 years ago
• ildoonet / pytorch-randaugment
  Unofficial PyTorch Reimplementation of RandAugment.
  ☆631Updated last year
• lucidrains / pixel-level-contrastive-learning
  Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch
  ☆257Updated 4 years ago
• teddykoker / torchsort
  Fast, differentiable sorting and ranking in PyTorch
  ☆792Updated last year
• ptrblck / pytorch_misc
  Code snippets created for the PyTorch discussion board
  ☆557Updated 4 years ago
• kekmodel / MPL-pytorch
  Unofficial PyTorch implementation of "Meta Pseudo Labels"
  ☆387Updated last year
• nocotan / pytorch-lightning-gans
  Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.
  ☆166Updated 2 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.
  ☆490Updated last year
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆327Updated last year
• NVlabs / NVAE
  The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)
  ☆1,038Updated 2 years ago
• lucidrains / byol-pytorch
  Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
  ☆1,795Updated 7 months ago