BIGBALLON / distribuuuu
The pure and clear PyTorch Distributed Training Framework.
☆275Updated last year
Alternatives and similar repositories for distribuuuu:
Users that are interested in distribuuuu are comparing it to the libraries listed below
- Simple tutorials on Pytorch DDP training☆272Updated 2 years ago
- Yet another PyTorch Trainer and some core components for deep learning.☆211Updated 8 months ago
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆230Updated 4 years ago
- 一款便捷的抢占显卡脚本☆298Updated last week
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆594Updated 5 years ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆91Updated 2 years ago
- A quickstart and benchmark for pytorch distributed training.☆1,650Updated 6 months ago
- Some tricks of pytorch...☆1,178Updated 7 months ago
- A brief of TorchScript by MNIST☆107Updated 2 years ago
- 📊 A simple command-line utility for querying and monitoring GPU status☆88Updated 2 years ago
- Code release for "LogME: Practical Assessment of Pre-trained Models for Transfer Learning" (ICML 2021) and Ranking and Tuning Pre-trained…☆205Updated last year
- [NeurIPS 2020] This project provides a strong single-stage baseline for Long-Tailed Classification, Detection, and Instance Segmentation …☆563Updated 5 months ago
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆357Updated last year
- Accelerate training by storing parameters in one contiguous chunk of memory.☆292Updated 4 years ago
- Masked Autoencoders Are Scalable Vision Learners☆247Updated last year
- Papers for normalization techniques, released codes collections.☆224Updated 4 years ago
- [ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?☆330Updated 2 years ago
- ☆195Updated 6 months ago
- pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同 方法的训练速度以及GPU内存的使用☆87Updated 10 months ago
- Code release for "Self-Tuning for Data-Efficient Deep Learning" (ICML 2021)☆110Updated 3 years ago
- RepMLPNet: Hierarchical Vision MLP with Re-parameterized Locality (CVPR 2022)☆306Updated last year
- pytorch memory track code☆1,013Updated 3 years ago
- [ICML 2022, Oral] The PyTorch Implementation of Adaptive Inertia Methods. The algorithms are based on our paper: "Adaptive Inertia: Dise…☆145Updated last year
- A general framework for inferring CNNs efficiently. Reduce the inference latency of MobileNet-V3 by 1.3x on an iPhone XS Max without sac…☆182Updated last year
- some tircks for PyTorch☆579Updated 5 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆185Updated 2 years ago
- Knowledge Distillation: CVPR2020 Oral, Revisiting Knowledge Distillation via Label Smoothing Regularization☆583Updated last year
- Distilling Knowledge via Knowledge Review, CVPR 2021☆267Updated 2 years ago
- [CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning☆286Updated 2 years ago
- 整理 pytorch 单机多 GPU 训练方法与原理☆795Updated 3 years ago