HongxinXiang / pytorch-multi-GPU-training-tutorial
☆65Updated 2 years ago
Alternatives and similar repositories for pytorch-multi-GPU-training-tutorial:
Users that are interested in pytorch-multi-GPU-training-tutorial are comparing it to the libraries listed below
- Yet another PyTorch Trainer and some core components for deep learning.☆211Updated 8 months ago
- pytorch单精度、半精度、混合精度、单卡、多卡(DP / DDP)、FSDP、DeepSpeed模型训练代码,并对比不同方法的训练速度以及GPU内存的使用☆87Updated 10 months ago
- The pure and clear PyTorch Distributed Training Framework.☆275Updated 11 months ago
- A light-weight script for maintaining a LOT of machine learning experiments.☆90Updated 2 years ago
- A brief of TorchScript by MNIST☆107Updated 2 years ago
- 多模态 MM +Chat 合集☆239Updated last week
- 这里是改进了pytorch的DataParallel, 用来平衡第一个GPU的显存使用量☆230Updated 4 years ago
- pytorch分布式训练,支持多机多卡,单机多卡。☆40Updated 3 years ago
- an implementation of transformer, bert, gpt, and diffusion models for learning purposes☆150Updated 3 months ago
- Simple tutorials on Pytorch DDP training☆272Updated 2 years ago
- 包含程序员面试大厂面试题和面试经验☆116Updated 3 weeks ago
- Pytorch Lightning入门中文教程,转载请注明来源。(当初是写着玩的,建议看完MNIST这个例子再上手)☆200Updated 4 years ago
- DeepSpeed Tutorial☆93Updated 5 months ago
- Official PyTorch implementation of training vision transformers with only 2040 images.☆52Updated 2 years ago
- DeepSpeed教程 & 示例注释 & 学习笔记 (大模型高效训练)☆146Updated last year
- A demo of image classification with PyTorch DDP (DistributedDataParallel) and AMP (Automatic Mixed Precision) modules.☆59Updated 10 months ago
- (Unofficial) PyTorch implementation of grouped-query attention (GQA) from "GQA: Training Generalized Multi-Query Transformer Models from …☆146Updated 8 months ago
- [CVPR 2023 Highlight] This is the official implementation of "Stitchable Neural Networks".☆246Updated last year
- Implementation of the Transformer variant proposed in "Transformer Quality in Linear Time"☆357Updated last year
- ☆60Updated 2 years ago
- 关于Transformer模型的最简洁pytorch实现,包含详细注释☆171Updated last year
- ☆120Updated last year
- Implementation of some unbalanced loss like focal_loss, dice_loss, DSC Loss, GHM Loss et.al☆253Updated last year
- 蜻蜓点论文 Think不Clear, 论文解读视频上传B站, youtube, 西瓜视频(同步到抖音)☆242Updated last year
- 整理 pytorch 单机多 GPU 训练方法与原理☆792Updated 3 years ago
- How to use wandb?☆603Updated last year
- Slides with modifications for a course at Tsinghua University.☆58Updated 2 years ago
- Model Compression 1. Pruning(BN Pruning) 2. Knowledge Distillation (Hinton) 3. Quantization (MNN) 4. Deployment (MNN)☆78Updated 4 years ago
- TensorRT 2022复赛方案: 首个基于Transformer的图像重建模型MST++的TensorRT模型推断优化☆137Updated 2 years ago
- ☆80Updated last year