katsura-jp / pytorch-cosine-annealing-with-warmup

Related projects: ⓘ

• fadel / pytorch_ema
  Tiny PyTorch library for maintaining a moving average of a collection of parameters.
  ☆400Updated 2 years ago
• Tony-Y / pytorch_warmup
  Learning Rate Warmup in PyTorch
  ☆387Updated 4 months ago
• xxxnell / how-do-vits-work
  (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"
  ☆806Updated 2 years ago
• ildoonet / pytorch-gradual-warmup-lr
  Gradually-Warmup Learning Rate Scheduler for PyTorch
  ☆971Updated 3 years ago
• kekmodel / MPL-pytorch
  Unofficial PyTorch implementation of "Meta Pseudo Labels"
  ☆383Updated 8 months ago
• lucidrains / mlp-mixer-pytorch
  An All-MLP solution for Vision, from Google AI
  ☆987Updated this week
• davda54 / sam
  SAM: Sharpness-Aware Minimization (PyTorch)
  ☆1,742Updated 6 months ago
• jacobgil / vit-explain
  Explainability for Vision Transformers
  ☆820Updated 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
• vballoli / nfnets-pytorch
  NFNets and Adaptive Gradient Clipping for SGD implemented in PyTorch. Find explanation at tourdeml.github.io/blog/
  ☆343Updated 8 months ago
• kakaobrain / torchlars
  A LARS implementation in PyTorch
  ☆332Updated 4 years ago
• facebookresearch / simsiam
  PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566
  ☆1,147Updated 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
• chinhsuanwu / coatnet-pytorch
  A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"
  ☆362Updated 2 years ago
• lucidrains / byol-pytorch
  Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
  ☆1,735Updated 2 months 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
• ildoonet / pytorch-randaugment
  Unofficial PyTorch Reimplementation of RandAugment.
  ☆627Updated last year
• SwinTransformer / Transformer-SSL
  This is an official implementation for "Self-Supervised Learning with Swin Transformers".
  ☆618Updated 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.
  ☆220Updated 7 months ago
• facebookresearch / msn
  Masked Siamese Networks for Label-Efficient Learning (https://arxiv.org/abs/2204.07141)
  ☆447Updated 2 years ago
• sthalles / PyTorch-BYOL
  PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning
  ☆476Updated 2 years ago
• lucidrains / perceiver-pytorch
  Implementation of Perceiver, General Perception with Iterative Attention, in Pytorch
  ☆1,078Updated last year
• facebookresearch / moco-v3
  PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057
  ☆1,201Updated 2 years ago
• qubvel / ttach
  Image Test Time Augmentation with PyTorch!
  ☆966Updated last year
• Spijkervet / SimCLR
  PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations by T. Chen et al.
  ☆752Updated 3 months ago
• lucidrains / linformer
  Implementation of Linformer for Pytorch
  ☆244Updated 8 months ago
• tatp22 / multidim-positional-encoding
  An implementation of 1D, 2D, and 3D positional encoding in Pytorch and TensorFlow
  ☆525Updated 8 months ago
• PatrickHua / SimSiam
  A pytorch implementation for paper 'Exploring Simple Siamese Representation Learning'
  ☆810Updated 2 years ago
• google-research / sam
  ☆552Updated last month