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
- CM-NAS: Cross-Modality Neural Architecture Search for Visible-Infrared Person Re-Identification (ICCV2021)☆48Aug 12, 2021Updated 4 years ago
- Code of our Neurips2020 paper "Auto Learning Attention", coming soon☆22Apr 14, 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 3 months ago
- Extend BoxDiff to SDXL (SDXL-based layout-to-image generation)☆27May 23, 2024Updated last year
- [IJCNN'19, IEEE JSTSP'19] Caffe code for our paper "Structured Pruning for Efficient ConvNets via Incremental Regularization"; [BMVC'18] …☆14Feb 14, 2020Updated 6 years ago
- ☆17Jul 10, 2022Updated 3 years ago
- Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks☆18Nov 5, 2019Updated 6 years ago
- Revisiting Parameter Sharing for Automatic Neural Channel Number Search, NeurIPS 2020☆22Nov 15, 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
- ☆13Sep 22, 2025Updated 5 months ago
- A Unified, Systematic Framework of Structured Weight Pruning for DNNs☆22Aug 3, 2018Updated 7 years ago
- Reducing Channel Redundancy in Convolutional Neural Networks by Features Recombining (TIP 2021)☆20Mar 1, 2023Updated 3 years ago
- Manually implemented quantization-aware training☆23Oct 12, 2022Updated 3 years ago
- Official implementation of the ICCV2023 paper: Enhancing Generalization of Universal Adversarial Perturbation through Gradient Aggregatio…☆28Aug 17, 2023Updated 2 years ago
- 深度学习OCR REST api (Flask+Redis+Keras)☆12Jul 23, 2018Updated 7 years ago
- easyanimate generete videos with ExLlamaV2 quantization LLM prompt☆13Jun 26, 2024Updated last year
- Code for our ICLR'2021 paper "DrNAS: Dirichlet Neural Architecture Search"☆43Apr 12, 2021Updated 4 years ago
- Repo with code for NIR'24 challange☆14Apr 22, 2024Updated last year
- CUDA project for uni subject☆26Oct 26, 2020Updated 5 years ago
- ☆13Feb 3, 2022Updated 4 years ago
- [ACCV 2024] Official code of our paper "Joint Image Super-resolution and Low-light Enhancement in the Dark"☆14Jun 23, 2025Updated 8 months ago
- Good Subnetworks Provably Exist: Pruning via Greedy Forward Selection☆21Jan 21, 2021Updated 5 years ago
- Code for the paper "Faster Neural Network Training with Approximate Tensor Operations"☆10Oct 23, 2021Updated 4 years ago
- [ICLR'21] Neural Pruning via Growing Regularization (PyTorch)☆82Jul 15, 2021Updated 4 years ago
- A template for PyTorch projects.☆22Aug 28, 2018Updated 7 years ago
- zombie game☆11Apr 19, 2019Updated 6 years ago
- [TPAMI 2021] DVG-Face: Dual Variational Generation for Heterogeneous Face Recognition☆77Nov 13, 2023Updated 2 years ago
- An implementation of ResNet with mixup and cutout regularizations and soft filter pruning.☆17Feb 23, 2026Updated 3 weeks ago
- PyTorch, PyTorch Lightning framework for trying knowledge distillation in image classification problems☆32Jul 25, 2024Updated last year
- ☆17Sep 6, 2025Updated 6 months ago
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19) https://arxiv.org/abs/1904.01355☆38Jan 6, 2020Updated 6 years ago
- [ICLR 2025] Linear Combination of Saved Checkpoints Makes Consistency and Diffusion Models Better☆16Feb 15, 2025Updated last year
- Efficient Scaling of Neurons for Resource-Constrained Deep Neural Networks☆25Jul 14, 2020Updated 5 years ago
- The collection of training tricks of binarized neural networks.☆72Apr 2, 2021Updated 4 years ago
- All the useful tools I have been using while working in data science for remote sensing☆11Nov 27, 2019Updated 6 years ago
- 基于 frp 使用 colab 的 ssh☆10Jul 10, 2020Updated 5 years ago