当前位置：网站首页>【科研-学习-pytorch】3-分类问题

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

2022-08-09 00:25:00 【helloworld_Fly】

分类问题

线性回归，三个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变化和部分预测结果；

原网站

版权声明
本文为[helloworld_Fly]所创，转载请带上原文链接，感谢
https://blog.csdn.net/helloworld_Fly/article/details/124873048

边栏推荐

猜你喜欢

随机推荐