PyTorch-小土堆跟学-入门保姆级教程

github链接及视频

https://github.com/xiaotudui/pytorch-tutorial

https://www.bilibili.com/video/av74281036/?vd_source=48334bf584e5051d1187b1f0f2d076e8

训练套路

准备数据集(原始数据集合；已标注的数据集)

处理数据集(原始数据集要做标注；已标注的数据集获取数据长度辅助计算精确度等)

加载数据集(DataLoader)

搭建网络模型

创建网络模型(类实例化)

定义损失函数、优化器

设置训练网络的参数(如：训练次数、轮数等等)

模型训练

模型保存

评估模型(测试数据集评估)

优化