xxcheng0708 / pytorch-model-train-template
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆77Updated 8 months ago
Related projects ⓘ
Alternatives and complementary repositories for pytorch-model-train-template
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆116Updated last year
- DeepSpeed Tutorial☆89Updated 3 months ago
- ☆64Updated 2 years ago
- Keras implement of Finite Scalar Quantization☆63Updated last year
- Yet another PyTorch Trainer and some core components for deep learning.☆206Updated 6 months ago
- ☆76Updated 4 months ago
- ☆154Updated last month
- A paper list of some recent works about Token Compress for Vit and VLM☆139Updated this week
- Efficient Multimodal Large Language Models: A Survey☆278Updated 3 months ago
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆149Updated 2 weeks ago
- Grab GPU whenever available☆279Updated 2 years ago
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆86Updated 6 months ago
- The pure and clear PyTorch Distributed Training Framework.☆275Updated 9 months ago
- 这是一个DiT-pytorch的代码,主要用于学习DiT结构。☆63Updated 8 months ago
- LoRAMoE: Revolutionizing Mixture of Experts for Maintaining World Knowledge in Language Model Alignment☆230Updated 6 months ago
- Lion and Adam optimization comparison☆56Updated last year
- ☆25Updated 3 weeks ago
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆185Updated last week
- 多模态 MM +Chat 合集☆207Updated 2 weeks ago
- ☆35Updated 5 months ago
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆244Updated last year
- The record of what I‘ve been through.☆95Updated last month
- ☆181Updated 11 months ago
- 和李沐一起读论文☆132Updated 3 weeks ago
- Pytorch Lightning入门中文教程,转载请注明来源。(当初是写着玩的,建议看完MNIST这个例子再上手)☆192Updated 3 years ago
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆133Updated 6 months ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆90Updated 2 years ago
- Harnessing 1.4M GPT4V-synthesized Data for A Lite Vision-Language Model☆246Updated 4 months ago
- Simple tutorials on Pytorch DDP training☆267Updated 2 years ago
- ☆220Updated 7 months ago