Learning Rate Warmup in PyTorch
☆412Jun 19, 2025Updated last year
Alternatives and similar repositories for pytorch_warmup
Users that are interested in pytorch_warmup are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆985Oct 10, 2024Updated 2 years ago
- ☆470Apr 8, 2023Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,172Mar 22, 2024Updated 2 years ago
- Permutation invariant training in PyTorch☆13Oct 2, 2020Updated 6 years ago
- Implementation of paper "Do Wide and Deep Networks Learn the Same Things?"☆16Mar 15, 2022Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆37,284Updated this week
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,207Dec 22, 2023Updated 2 years ago
- A learning rate range test implementation in PyTorch☆1,009Jun 24, 2025Updated last year
- Chainer implementation of CIFAR-10 dataset training☆12Dec 7, 2022Updated 3 years ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,984Feb 21, 2024Updated 2 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,548Jul 31, 2021Updated 5 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,083Jul 8, 2024Updated 2 years ago
- neural network based speaker embedder☆24Jan 7, 2023Updated 3 years ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,207Jul 9, 2026Updated 4 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆25,571Nov 7, 2026Updated last week
- Mila technical documentation☆13Updated this week
- PyTorch implementation of some learning rate schedulers for deep learning researcher.☆90Sep 6, 2022Updated 4 years ago
- A new architecture of semantic segmentation called Dense-Attention Networks.☆14Nov 10, 2018Updated 8 years ago
- Deploy KoGPT with Triton Inference Server☆14Nov 18, 2022Updated 3 years ago
- ☆13Oct 8, 2021Updated 5 years ago
- optimizer & lr scheduler & loss function collections in PyTorch☆436Oct 30, 2026Updated 2 weeks ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆825Jun 8, 2025Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,634Aug 26, 2026Updated 2 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆413Jan 13, 2021Updated 5 years ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,942Updated this week
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,254Aug 19, 2026Updated 2 months ago
- Code release for ConvNeXt model☆6,410Jan 8, 2023Updated 3 years ago
- ☆10Mar 1, 2018Updated 8 years ago
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆31,449Oct 30, 2026Updated 2 weeks ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Jul 31, 2023Updated 3 years ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆52Jun 26, 2025Updated last year
- Vector (and Scalar) Quantization, in Pytorch☆4,028Updated this week
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- View model summaries in PyTorch!☆2,950Updated this week
- Utility functions that I reuse across different projects☆15Jun 4, 2021Updated 5 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆229Mar 12, 2026Updated 8 months ago
- An implementation of capsule routing for sound event detection☆15Jan 29, 2019Updated 7 years ago
- Code release for ConvNeXt V2 model☆2,066Aug 14, 2024Updated 2 years ago
- Warmup learning rate wrapper for Pytorch Scheduler☆42Apr 12, 2020Updated 6 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆13,013Nov 1, 2026Updated last week
Are these results useful?