1--数据载入
|----流程: DataSet->DataLoader->调用DataLoader
|----DataLoader迭代器读不到数据,无报错,一直卡住的显现:
DataLoader的num_worker设置为0 解决,这个多线程读取数据不知道为什么会出现这个bug?
2--定义网络
|----与之前的一致
3--损失函数和优化器
|----损失函数: 交叉熵
|----优化器:SGD (?)
扫描二维码关注公众号,回复:
5342052 查看本文章
4--网络训练
|----读入数据
|----输入网络
|----获得结果,计算loss
|----清空梯度
|----反向传播
|--训练结束