Learning Rate Warmup in PyTorch
☆415Jun 19, 2025Updated 8 months ago
Alternatives and similar repositories for pytorch_warmup
Users that are interested in pytorch_warmup are comparing it to the libraries listed below
Sorting:
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆991Oct 10, 2024Updated last year
- ☆467Apr 8, 2023Updated 2 years ago
- A learning rate range test implementation in PyTorch☆1,002Jun 24, 2025Updated 8 months ago
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,207Dec 22, 2023Updated 2 years ago
- ☆13Oct 8, 2021Updated 4 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,963Feb 21, 2024Updated 2 years ago
- Deploy KoGPT with Triton Inference Server☆14Nov 18, 2022Updated 3 years ago
- On the Variance of the Adaptive Learning Rate and Beyond☆2,549Jul 31, 2021Updated 4 years ago
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆227Dec 16, 2025Updated 2 months ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,461Updated this week
- neural network based speaker embedder☆25Jan 7, 2023Updated 3 years ago
- Code release for ConvNeXt model☆6,302Jan 8, 2023Updated 3 years ago
- Utility functions that I reuse across different projects☆14Jun 4, 2021Updated 4 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Jul 31, 2023Updated 2 years ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,528Updated this week
- Implementation of Online Label Smoothing in PyTorch☆96Sep 27, 2022Updated 3 years ago
- A new architecture of semantic segmentation called Dense-Attention Networks.☆14Nov 10, 2018Updated 7 years ago
- An implementation of capsule routing for sound event detection☆15Jan 29, 2019Updated 7 years ago
- This is the official implementation of "Hard-aware Instance Adaptive Self-training for Unsupervised Cross-domain Semantic Segmentation".☆15Mar 29, 2025Updated 11 months ago
- Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models☆808Jun 8, 2025Updated 9 months ago
- AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights (ICLR 2021)☆415Jan 13, 2021Updated 5 years ago
- Framework to train semantic segmentation models on Pytorch using yaml config files☆17Feb 28, 2026Updated last week
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,664Apr 7, 2025Updated 11 months ago
- PyTorch implementation of some learning rate schedulers for deep learning researcher.☆91Sep 6, 2022Updated 3 years ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,225Feb 25, 2026Updated last week
- A Track-Wise Ensemble Event Independent Network for 3D Polyphonic Sound Event Localization and Detection☆22Nov 14, 2024Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,415Feb 20, 2026Updated 2 weeks ago
- Machine learning metrics for distributed, scalable PyTorch applications.☆2,412Updated this week
- Code release for ConvNeXt V2 model☆1,981Aug 14, 2024Updated last year
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,906Updated this week
- Count the MACs / FLOPs of your PyTorch model.☆5,081Jul 8, 2024Updated last year
- Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"☆1,068Aug 9, 2024Updated last year
- Ranger deep learning optimizer rewrite to use newest components☆341Feb 18, 2024Updated 2 years ago
- View model summaries in PyTorch!☆2,903Mar 2, 2026Updated last week
- Image Test Time Augmentation with PyTorch!☆1,028Jul 28, 2023Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,260Oct 17, 2024Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,471May 2, 2025Updated 10 months ago
- PyTorch Lightning + Hydra. A very user-friendly template for ML experimentation. ⚡🔥⚡☆5,179Aug 16, 2024Updated last year