Pytorch ImageNet training codes with various tricks, lr schedulers, distributed training, mixed precision training, DALI dataloader etc.
☆18Aug 12, 2020Updated 5 years ago
Alternatives and similar repositories for ImageNet-training
Users that are interested in ImageNet-training are comparing it to the libraries listed below
Sorting:
- TF-NAS: Rethinking Three Search Freedoms of Latency-Constrained Differentiable Neural Architecture Search (ECCV2020)☆75May 23, 2021Updated 4 years ago
- Official codebase for our paper "Joslim: Joint Widths and Weights Optimization for Slimmable Neural Networks"☆12Jun 30, 2021Updated 4 years ago
- Code for "Fast Sparse ConvNets" CVPR2020 submissions☆12Nov 20, 2019Updated 6 years ago
- pruning vision models in torch☆17Dec 5, 2025Updated 2 months ago
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks☆18Nov 5, 2019Updated 6 years ago
- ☆17Jul 10, 2022Updated 3 years ago
- Experiments from "The Generalization-Stability Tradeoff in Neural Network Pruning": https://arxiv.org/abs/1906.03728.☆14Oct 23, 2020Updated 5 years ago
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2021 -- Network Pruning using Adaptive Exemplar Filters☆24Apr 4, 2021Updated 4 years ago
- Reducing Channel Redundancy in Convolutional Neural Networks by Features Recombining (TIP 2021)☆20Mar 1, 2023Updated 3 years ago
- An implementation of ResNet with mixup and cutout regularizations and soft filter pruning.☆17Updated this week
- Revisiting Parameter Sharing for Automatic Neural Channel Number Search, NeurIPS 2020☆22Nov 15, 2020Updated 5 years ago
- A Unified, Systematic Framework of Structured Weight Pruning for DNNs☆22Aug 3, 2018Updated 7 years ago
- Implementation of PGONAS for CVPR22W and RD-NAS for ICASSP23☆23Apr 25, 2023Updated 2 years ago
- Efficient Scaling of Neurons for Resource-Constrained Deep Neural Networks☆25Jul 14, 2020Updated 5 years ago
- Code of our Neurips2020 paper "Auto Learning Attention", coming soon☆22Apr 14, 2021Updated 4 years ago
- Good Subnetworks Provably Exist: Pruning via Greedy Forward Selection☆21Jan 21, 2021Updated 5 years ago
- A template for PyTorch projects.☆22Aug 28, 2018Updated 7 years ago
- ☆25Dec 11, 2021Updated 4 years ago
- PyTorch, PyTorch Lightning framework for trying knowledge distillation in image classification problems☆31Jul 25, 2024Updated last year
- This repo is about NAS☆26Nov 1, 2019Updated 6 years ago
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2022 -- Distilling a Powerful Student Model via Online Knowledge Distillation☆31Nov 11, 2021Updated 4 years ago
- ☆68Nov 5, 2020Updated 5 years ago
- ZSKD with PyTorch☆31Jun 26, 2023Updated 2 years ago
- Code for BlockSwap (ICLR 2020).☆33Mar 25, 2021Updated 4 years ago
- Codes for paper "Few Shot Network Compression via Cross Distillation", AAAI 2020.☆31Jan 31, 2020Updated 6 years ago
- Pytorch implementation of our paper accepted by TPAMI 2023 — Lottery Jackpots Exist in Pre-trained Models☆35Jun 19, 2023Updated 2 years ago
- The collection of training tricks of binarized neural networks.☆72Apr 2, 2021Updated 4 years ago
- [TPAMI 2021] DVG-Face: Dual Variational Generation for Heterogeneous Face Recognition☆77Nov 13, 2023Updated 2 years ago
- Code for reproducing work of ICML 2019 paper: Memory-Optimal Direct Convolutions for Maximizing Classification Accuracy in Embedded Appli…☆12Jun 8, 2019Updated 6 years ago
- Reinforcement learning modular with pytorch☆11Jan 18, 2021Updated 5 years ago
- A tracing tool to analyze the I/O behavior of a program.☆12Sep 25, 2019Updated 6 years ago
- ☆11Apr 17, 2021Updated 4 years ago
- Paper survey of efficient computation for large scale models.☆34Dec 7, 2024Updated last year
- All about acceleration and compression of Deep Neural Networks☆33Nov 5, 2019Updated 6 years ago
- Generalized Framework for PyTorch☆32Nov 25, 2021Updated 4 years ago
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2021 -- Filter Sketch for Network Pruning☆53Feb 11, 2021Updated 5 years ago
- Experiments with the ideas presented in https://arxiv.org/abs/2003.00152 by Frankle et al.☆29Aug 21, 2020Updated 5 years ago
- pytorch fixed point training tool/framework☆34Oct 14, 2020Updated 5 years ago
- Code repositoy for "AOWS: Adaptive and optimal network width search with latency constraints", CVPR 2020☆36Jun 19, 2020Updated 5 years ago