lhangggg / PytorchPipeline

Pytorch pipeline template

PytorchPipeline

本教程《从零搭建pytorch模型教程》来自公众号和知识星球【CV技术指南】。未经许可，不得用于商业行为。

关于Pytorch每个部分的写法，使用，可以在QQ群 （444129970）里下载《从零搭建pytorch模型教程》pdf。或者点击以下链接。

代码已经更新，后续将进一步新增一些新的功能。如果日常使用中出现什么问题，都可以在群里提出，我们将尽快改进。

后续更新计划

1. 新增一个没有多机多卡和混合精度训练的train文件。
2. 新增各种自定义损失函数。
3. 新增各种网络结构。

代码教程

从零搭建Pytorch模型教程（一）数据读取

从零搭建Pytorch模型教程（二）搭建网络

从零搭建Pytorch模型教程（三）搭建Transformer网络

从零搭建Pytorch模型教程（四）编写训练过程--参数解析

从零搭建Pytorch模型教程（五）编写训练过程--一些基本的配置

从零搭建Pytorch模型教程（六）编写训练过程和推理过程

从零搭建Pytorch模型教程（七）单机多卡和多机多卡训练

从零搭建pytorch模型教程（八）实践部分（一）微调、冻结网络

从零搭建pytorch模型教程（八）实践部分（二）目标检测数据集格式转换

一份热力图可视化代码使用教程

一份可视化特征图的代码

其余一些文章说明请下载文档。