Pytorch ImageNet training codes with various tricks, lr schedulers, distributed training, mixed precision training, DALI dataloader etc.
☆18Aug 12, 2020Updated 6 years ago
Alternatives and similar repositories for ImageNet-training
Users that are interested in ImageNet-training are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- TF-NAS: Rethinking Three Search Freedoms of Latency-Constrained Differentiable Neural Architecture Search (ECCV2020)☆75May 23, 2021Updated 5 years ago
- Code of our Neurips2020 paper "Auto Learning Attention", coming soon☆22Apr 14, 2021Updated 5 years ago
- Official codebase for our paper "Joslim: Joint Widths and Weights Optimization for Slimmable Neural Networks"☆12Jun 30, 2021Updated 5 years ago
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆31Oct 31, 2019Updated 6 years ago
- Code for "Fast Sparse ConvNets" CVPR2020 submissions☆12Nov 20, 2019Updated 6 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.
- ☆11Dec 1, 2020Updated 5 years ago
- 😎 Awesome lists of papers and codes about Large Vision-Language Models☆13Apr 1, 2024Updated 2 years ago
- ☆15Jan 8, 2020Updated 6 years ago
- ☆17Jul 10, 2022Updated 4 years ago
- Experiments from "The Generalization-Stability Tradeoff in Neural Network Pruning": https://arxiv.org/abs/1906.03728.☆14Oct 23, 2020Updated 6 years ago
- Revisiting Parameter Sharing for Automatic Neural Channel Number Search, NeurIPS 2020☆21Nov 15, 2020Updated 5 years ago
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks☆18Nov 5, 2019Updated 6 years ago
- Video joint denoising and demosaicing with recurrent CNNs - WACV 2023☆13Sep 19, 2023Updated 3 years ago
- A Unified, Systematic Framework of Structured Weight Pruning for DNNs☆22Aug 3, 2018Updated 8 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- The official code of paper "OMS-DPM: Optimizing Model Schedule for Diffusion Probabilistic Model" accepted by ICML 2023☆24Oct 11, 2023Updated 3 years ago
- Manually implemented quantization-aware training☆22Oct 12, 2022Updated 4 years ago
- Code for our ICLR'2021 paper "DrNAS: Dirichlet Neural Architecture Search"☆43Apr 12, 2021Updated 5 years ago
- CNN Visualization Grad-CAM☆10Aug 5, 2020Updated 6 years ago
- [ACCV 2024] Official code of our paper "Joint Image Super-resolution and Low-light Enhancement in the Dark"☆14Jun 23, 2025Updated last year
- ☆13Sep 22, 2025Updated last year
- GGUF implementation for the ComfyUI Ultimate SD Upscale node.☆15Aug 22, 2025Updated last year
- Code for the paper "Faster Neural Network Training with Approximate Tensor Operations"☆10Oct 23, 2021Updated 5 years ago
- A template for PyTorch projects.☆23Aug 28, 2018Updated 8 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Project for SignalHound Spectrum Analyzer in Qt☆12May 4, 2017Updated 9 years ago
- [ICLR'21] Neural Pruning via Growing Regularization (PyTorch)☆82Jul 15, 2021Updated 5 years ago
- [TPAMI 2021] DVG-Face: Dual Variational Generation for Heterogeneous Face Recognition☆76Nov 13, 2023Updated 2 years ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19) https://arxiv.org/abs/1904.01355☆38Jan 6, 2020Updated 6 years ago
- pruning vision models in torch☆17Dec 5, 2025Updated 10 months ago
- An implementation of ResNet with mixup and cutout regularizations and soft filter pruning.☆17Apr 1, 2026Updated 6 months ago
- ☆14Feb 3, 2022Updated 4 years ago
- Implementation of PGONAS for CVPR22W and RD-NAS for ICASSP23☆23Apr 25, 2023Updated 3 years ago
- Quantize pytorch model, support post-training quantization and quantization aware training methods☆15Jun 15, 2023Updated 3 years 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.
- Megatron-LM setup in the smol-cluster☆12Aug 26, 2024Updated 2 years ago
- All the useful tools I have been using while working in data science for remote sensing☆11Nov 27, 2019Updated 6 years ago
- Interspeech Tutorial - Resource Efficient and Cross-Modal Learning Toward Foundation Modeling☆15Oct 9, 2023Updated 3 years ago
- This repository contains the code for my home improvements projects (home automation, plants watering, everyday calendar, motorized mobil…☆10Dec 9, 2022Updated 3 years ago
- 阿里云第二届数据库大赛新手门槛队（季军）解决方案☆10Apr 19, 2021Updated 5 years ago
- ☆12Jun 28, 2018Updated 8 years ago
- The collection of training tricks of binarized neural networks.☆72Apr 2, 2021Updated 5 years ago