昇思25天学习打卡营第7天|munger85

模型训练

构建数据集

在这里插入图片描述
数据集的准备我绝对是最重要
但是操作无非是处理,然后哪些是X哪些是Y,batch是多少。
在这里插入图片描述
神经网络的定义,前面的课程也有
唯一就是loss_fn = nn.CrossEntropyLoss()是定义损失函数,一般这个比较常见
根据任务的不同用不同的
回归任务的nn.MSELoss(均方误差)
用于分类的nn.NLLLoss(负对数似然)
最后通过训练,模型的参数不断优化的工具叫做优化器,优化器里会自动根据梯度去优化那些权重
optimizer = nn.SGD(model.trainable_params(), learning_rate=learning_rate) SGD是很常见的优化器
然后使用上次的自动微分,我们可以得到梯度,再用优化器把这些梯度利用了,去优化nn

在这里插入图片描述
定义个测试的
在这里插入图片描述

每训练一下就测一下准确度如何
在这里插入图片描述
这样在每一轮的训练后会出现1测试准确率,知道随着时间越多,训练越多,越来越准确
在这里插入图片描述

  • 3
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值