【科研-学习-pytorch】3-分类问题

分类问题

线性回归，三个fc分类预测；
one-hot编码；
激活函数Relu；
矩阵相乘的shape变换；

mnist

torchvision直接读取；
train、test、val；

实现

• load data
  torchvision load data（x，y）

• create model
  torch.nn，使用nn.linear嵌套三个fc得到模型；
  加入F.relu作为激活函数；

• train和test
  训练迭代，读取traindataloader数据、放入模型、清零梯度、计算梯度、反向传播；

• visual and predict result
  plot loss变化和部分预测结果；