BIGBALLON / distribuuuu
The pure and clear PyTorch Distributed Training Framework.
☆275Updated 9 months ago
Related projects ⓘ
Alternatives and complementary repositories for distribuuuu
- Simple tutorials on Pytorch DDP training☆267Updated 2 years ago
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆230Updated 4 years ago
- Yet another PyTorch Trainer and some core components for deep learning.☆206Updated 6 months ago
- Grab GPU whenever available☆279Updated 2 years ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆90Updated 2 years ago
- A quickstart and benchmark for pytorch distributed training.☆1,640Updated 3 months ago
- PyTorch Project Specification.☆667Updated 3 years ago
- pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用☆77Updated 8 months ago
- Some tricks of pytorch...☆1,160Updated 5 months ago
- Masked Autoencoders Are Scalable Vision Learners☆247Updated last year
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆350Updated last year
- Distilling Knowledge via Knowledge Review, CVPR 2021☆261Updated last year
- Sublinear memory optimization for deep learning. https://arxiv.org/abs/1604.06174☆589Updated 4 years ago
- A general framework for inferring CNNs efficiently. Reduce the inference latency of MobileNet-V3 by 1.3x on an iPhone XS Max without sac…☆181Updated last year
- [CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning☆284Updated 2 years ago
- [ICLR 2021 top 3%] Is Attention Better Than Matrix Decomposition?☆325Updated 2 years ago
- ☆191Updated 3 years ago
- Papers for normalization techniques, released codes collections.☆225Updated 4 years ago
- Code release for "LogME: Practical Assessment of Pre-trained Models for Transfer Learning" (ICML 2021) and Ranking and Tuning Pre-trained…☆203Updated last year
- Official MegEngine implementation of RepLKNet☆268Updated 2 years ago
- [ICLR 2022] Official implementation of cosformer-attention in cosFormer: Rethinking Softmax in Attention☆179Updated last year
- (CVPR 2021, Oral) Dynamic Slimmable Network☆228Updated 2 years ago
- Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"☆425Updated last year
- RepMLPNet: Hierarchical Vision MLP with Re-parameterized Locality (CVPR 2022)☆303Updated last year
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆244Updated last year
- [NeurIPS 2021] [T-PAMI] DynamicViT: Efficient Vision Transformers with Dynamic Token Sparsification☆572Updated last year
- CVPR2022, BatchFormer: Learning to Explore Sample Relationships for Robust Representation Learning, https://arxiv.org/abs/2203.01522☆248Updated last year
- Accelerate training by storing parameters in one contiguous chunk of memory.☆291Updated 4 years ago
- 整理 pytorch 单机多 GPU 训练方法与原理☆775Updated 2 years ago
- PASSL包含 SimCLR,MoCo v1/v2,BYOL,CLIP,PixPro,simsiam, SwAV, BEiT,MAE 等图像自监督算法以及 Vision Transformer,DEiT,Swin Transformer,CvT,T2T-ViT,MLP-…☆276Updated last year