xxcheng0708 / pytorch-model-train-template
pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用
☆87Updated 11 months ago
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教程 & 示例注释 & 学习笔记 (大模型高效训练)☆150Updated last year
- DeepSpeed Tutorial☆95Updated 6 months ago
- Yet another PyTorch Trainer and some core components for deep learning.☆211Updated 9 months ago
- ☆177Updated 4 months ago
- 主要记录大语言大模型(LLMs) 算法(应用)工程师多模态相关知识☆121Updated 9 months ago
- The pure and clear PyTorch Distributed Training Framework.☆276Updated last year
- Keras implement of Finite Scalar Quantization☆69Updated last year
- 和李沐一起读论文☆167Updated last month
- 一款便捷的抢占显卡脚本☆300Updated last month
- Reading notes about Multimodal Large Language Models, Large Language Models, and Diffusion Models☆250Updated this week
- A generalized framework for subspace tuning methods in parameter efficient fine-tuning.☆128Updated 2 weeks ago
- My implementation of "Patch n’ Pack: NaViT, a Vision Transformer for any Aspect Ratio and Resolution"☆213Updated 3 weeks ago
- ☆62Updated this week
- [NeurIPS 2024] Classification Done Right for Vision-Language Pre-Training☆200Updated last month
- ☆64Updated 2 years ago
- The record of what I‘ve been through.☆95Updated last month
- Simple tutorials on Pytorch DDP training☆273Updated 2 years ago
- ☆93Updated 7 months ago
- 多模态 MM +Chat 合集☆240Updated this week
- A brief of TorchScript by MNIST☆107Updated 2 years ago
- [Survey] Next Token Prediction Towards Multimodal Intelligence: A Comprehensive Survey☆357Updated last month
- ☆274Updated last week
- Train InternViT-6B in MMSegmentation and MMDetection with DeepSpeed☆79Updated 3 months ago
- an implementation of transformer, bert, gpt, and diffusion models for learning purposes☆151Updated 4 months ago
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆248Updated last year
- Implementation of "Attention Is Off By One" by Evan Miller☆189Updated last year
- Lion and Adam optimization comparison☆57Updated last year
- PyTorch Dataset Rank Dataset☆41Updated 3 years ago
- WWW2025 Multimodal Intent Recognition for Dialogue Systems Challenge☆114Updated 3 months ago
- ☆187Updated last year