#多GPU训练
deep-person-reid - 深度学习人员重识别库,支持多GPU训练和跨数据集评估
TorchreidPyTorch深度学习重识别多GPU训练Github开源项目
Torchreid是一个基于PyTorch的深度学习人员重识别库,支持多GPU训练、图像和视频重识别、端到端训练与评估、多数据集训练和跨数据集评估。它易于准备数据集,支持添加模型、数据集和训练方法,提供预训练模型和高级训练技术,并配备可视化工具。
wavegrad - 由Google Brain设计的高效神经声码器
WaveGrad神经声码器高质量合成多GPU训练预训练模型Github开源项目
WaveGrad是由Google Brain团队设计的神经声码器,专业于将对数缩放的Mel频谱图转换为波形。此项目提供稳定训练、合成、混合精度训练以及多GPU支持,且支持命令行和API推理接口,配备预训练模型。易于安装,支持通过pip和GitHub进行安装。其多GPU训练和混合精度训练提升了处理效率,适合需进行高效音频处理的开发者。
urban_seg - 针对初学者的遥感图片语义分割项目
urban_seg遥感图片语义分割unicom模型多GPU训练Github开源项目
一个针对初学者的遥感图片语义分割项目,使用在4亿张图片上预训练的unicom模型。该模型在遥感分割中表现出色,仅需4张图片训练即可取得良好效果。提供简单的单GPU和多GPU训练代码,帮助快速上手并提升性能。
Trainer - 基于PyTorch的通用模型训练框架
TrainerPyTorch模型训练多GPU训练实验日志Github开源项目
Trainer是一个基于PyTorch的开源模型训练框架,具有简洁的代码结构和灵活的优化控制。该框架支持自动优化、高级优化循环、批量大小查找、分布式训练和Accelerate集成。此外,Trainer提供回调功能、性能分析和多种实验日志记录选项,包括Tensorboard和ClearML等。这个框架适用于各类深度学习任务,能够简化训练流程并提升效率。
相关文章