欢迎来到我的博客!
1.PyTorch模型训练实用教程
https://github.com/TingsongYu/PyTorch_Tutorial
注:该教程主要内容为利用PyTorch训练模型可能涉及到的方法及函数,包括数据增强方法(22个)、权值初始化方法(10个)、损失函数(17个)、优化器(6个)及tensorboardx的方法(13个)。
主要目录如下
第一章 | 数据的划分、预处理及增强 |
第二章 | 模型的定义、权值初始化及模型Finetune |
第三章 | 各种损失函数及优化器 |
第四章 | 可视化工具,用于看看数据、模型及损失函数 |