xxcheng0708 / pytorch-model-train-template
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆98Updated last year
Alternatives and similar repositories for pytorch-model-train-template:
Users that are interested in pytorch-model-train-template are comparing it to the libraries listed below
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆159Updated last year
- DeepSpeed Tutorial☆96Updated 8 months ago
- ☆38Updated 3 months ago
- 多模态 MM +Chat 合集☆260Updated this week
- ☆194Updated 6 months ago
- [NeurIPS 2024] Classification Done Right for Vision-Language Pre-Training☆205Updated last month
- pytorch distribute tutorials☆126Updated last week
- ☆100Updated 10 months ago
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆401Updated 3 weeks ago
- Yet another PyTorch Trainer and some core components for deep learning.☆217Updated last year
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆165Updated 11 months ago
- an implementation of transformer, bert, gpt, and diffusion models for learning purposes☆155Updated 6 months ago
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆185Updated 11 months ago
- Efficient Multimodal Large Language Models: A Survey☆343Updated this week
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆229Updated last month
- Open-Qwen2VL: Compute-Efficient Pre-Training of Fully-Open Multimodal LLMs on Academic Resources☆190Updated last month
- The pure and clear PyTorch Distributed Training Framework.☆276Updated last year
- Personal Project: MPP-Qwen14B & MPP-Qwen-Next(Multimodal Pipeline Parallel based on Qwen-LM). Support [video/image/multi-image] {sft/conv…☆442Updated last month
- ☆315Updated 2 months ago
- Keras implement of Finite Scalar Quantization☆71Updated last year
- 和李沐一起读论文☆192Updated 3 months ago
- 这是一个DiT-pytorch的代码,主要用于学习DiT结构。☆75Updated last year
- Rethinking RL Scaling for Vision Language Models: A Transparent, From-Scratch Framework and Comprehensive Evaluation Scheme☆120Updated 3 weeks ago
- Lion and Adam optimization comparison☆61Updated 2 years ago
- Lightning Attention-2: A Free Lunch for Handling Unlimited Sequence Lengths in Large Language Models☆282Updated 2 months ago
- [Survey] Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey☆422Updated 3 months ago
- Lossless Training Speed Up by Unbiased Dynamic Data Pruning☆334Updated 7 months ago
- A brief of TorchScript by MNIST☆111Updated 2 years ago
- 本项目提供了基于910B的huggingface LLM模型的Tensor Parallel(TP)部署教程,同时也可以作为一份极简的TP学习代码。☆22Updated 7 months ago
- ☆183Updated 3 weeks ago