xxcheng0708 / pytorch-model-train-templateLinks
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆115Updated 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
Sorting:
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆176Updated last year
- ☆44Updated 6 months ago
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆227Updated last year
- DeepSpeed Tutorial☆101Updated last year
- Yet another PyTorch Trainer and some core components for deep learning.☆224Updated last year
- [COLM 2025] Open-Qwen2VL: Compute-Efficient Pre-Training of Fully-Open Multimodal LLMs on Academic Resources☆251Updated 3 months ago
- 一款便捷的抢占显卡脚本☆350Updated 7 months ago
- ☆69Updated 2 years ago
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆567Updated last month
- ☆206Updated 10 months ago
- 多模态 MM +Chat 合集☆275Updated this week
- 这是一个DiT-pytorch的代码,主要用于学习DiT结构。☆80Updated last year
- [Survey] Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey☆447Updated 7 months ago
- Collect the awesome works evolved around reasoning models like O1/R1 in visual domain☆41Updated last month
- A generalized framework for subspace tuning methods in parameter efficient fine-tuning.☆153Updated last month
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆250Updated last month
- Simple tutorials on Pytorch DDP training☆281Updated 3 years ago
- 和李沐一起读论文☆208Updated 2 months ago
- GPG: A Simple and Strong Reinforcement Learning Baseline for Model Reasoning☆159Updated 3 months ago
- ☆513Updated last month
- The pure and clear PyTorch Distributed Training Framework.☆274Updated last year
- Efficient Multimodal Large Language Models: A Survey☆365Updated 3 months ago
- ☆82Updated 3 months ago
- [NeurIPS 24] MoE Jetpack: From Dense Checkpoints to Adaptive Mixture of Experts for Vision Tasks☆129Updated 9 months ago
- Lion and Adam optimization comparison☆63Updated 2 years ago
- [NeurIPS 2024] Classification Done Right for Vision-Language Pre-Training☆213Updated 5 months ago
- Rethinking RL Scaling for Vision Language Models: A Transparent, From-Scratch Framework and Comprehensive Evaluation Scheme☆138Updated 4 months ago
- Cool Papers - Immersive Paper Discovery☆596Updated 2 months ago
- 本项目提供了基于910B的huggingface LLM模型的Tensor Parallel(TP)部署教程,同时也可以作为一份极简的TP学习代码。☆27Updated 11 months ago
- Diffusion Transformers (DiTs) trained on MNIST dataset☆125Updated last year