fadel / pytorch_ema

Related projects ⓘ

Alternatives and complementary repositories for pytorch_ema

• katsura-jp / pytorch-cosine-annealing-with-warmup
  ☆440Updated last year
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆392Updated this week
• lucidrains / ema-pytorch
  A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model
  ☆517Updated last month
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆335Updated 4 years ago
• imbue-ai / self_supervised
  A Pytorch-Lightning implementation of self-supervised algorithms
  ☆536Updated 2 years ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆977Updated last month
• 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
  ☆192Updated 6 months ago
• facebookresearch / barlowtwins
  PyTorch implementation of Barlow Twins.
  ☆968Updated 2 years ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆257Updated 10 months ago
• microsoft / esvit
  EsViT: Efficient self-supervised Vision Transformers
  ☆408Updated last year
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,097Updated last year
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆1,003Updated 2 months ago
• sthalles / PyTorch-BYOL
  PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
  ☆479Updated 2 years ago
• benjs / nfnets_pytorch
  Pre-trained NFNets with 99% of the accuracy of the official paper "High-Performance Large-Scale Image Recognition Without Normalization".
  ☆159Updated 3 years ago
• google-research / sam
  ☆566Updated 3 weeks ago
• facebookresearch / msn
  Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
  ☆449Updated 2 years ago
• SamsungLabs / ASAM
  Implementation of ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networks, ICML 2021.
  ☆140Updated 3 years ago
• assafshocher / ResizeRight
  The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).
  ☆548Updated last year
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆345Updated 10 months ago
• lucidrains / linear-attention-transformer
  Transformer based on a variant of attention that is linear complexity in respect to sequence length
  ☆698Updated 6 months ago
• PistonY / torch-toolbox
  🛠 Toolbox to extend PyTorch functionalities
  ☆417Updated 6 months ago
• fkodom / fft-conv-pytorch
  Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.
  ☆477Updated last year
• lessw2020 / Ranger21
  Ranger deep learning optimizer rewrite to use newest components
  ☆323Updated 9 months ago
• locuslab / convmixer
  Implementation of ConvMixer for "Patches Are All You Need? 🤷"
  ☆1,062Updated 2 years ago
• kekmodel / MPL-pytorch
  Unofficial PyTorch implementation of "Meta Pseudo Labels"
  ☆383Updated 10 months ago
• AntixK / PyTorch-Model-Compare
  Compare neural networks by their feature similarity
  ☆343Updated last year
• chingyaoc / DCL
  NeurIPS 2020, Debiased Contrastive Learning
  ☆283Updated last year
• nocotan / pytorch-lightning-gans
  Collection of PyTorch Lightning implementations of Generative Adversarial Network varieties presented in research papers.
  ☆167Updated 2 years ago
• wzlxjtu / PositionalEncoding2D
  A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.
  ☆250Updated 4 years ago