深度学习案例教程 课件3.4.3 手写数字识别网络训练_第1页
深度学习案例教程 课件3.4.3 手写数字识别网络训练_第2页
深度学习案例教程 课件3.4.3 手写数字识别网络训练_第3页
深度学习案例教程 课件3.4.3 手写数字识别网络训练_第4页
深度学习案例教程 课件3.4.3 手写数字识别网络训练_第5页
已阅读5页,还剩2页未读 继续免费阅读

付费下载

下载本文档

版权说明:本文档由用户提供并上传,收益归属内容提供方,若内容存在侵权,请进行举报或认领

文档简介

实践任务3-手写数字识别网络训练

模型训练

为了方便后续使用模型,可以将训练过程写成一个函数,向该函数传入网络模型、损失函数、优化器等必要对象后,在MNIST数据集上进行训练并打印日志观察过程。实践任务3-手写数字识别网络训练

模型训练deftrain(epoches):

forepochinrange(epoches):

fori,datainenumerate(train_loader):

(images,labels)=data

images=images.reshape(-1,28*28)

output=model(images)实践任务3-手写数字识别网络训练

模型训练deftrain(epoches):

forepochinrange(epoches):

fori,datainenumerate(train_loader):

(images,labels)=data

images=images.reshape(-1,28*28)

output=model(images)

#前向传播,计算损失,方向传播,更新参数

loss=criterion(output,labels)

optimizer.zero_grad()

loss.backward()

optimizer.step()

实践任务3-手写数字识别网络训练

模型训练deftrain(epoches):

forepochinrange(epoches):

fori,datainenumerate(train_loader):

(images,labels)=data

#得到输入和标签

#前向传播,计算损失,方向传播,更新参数

#打印日志

if(i+1)%100==0:

print('Epoch[{}/{}-{}],Loss:{:.4f}'

.format(epoch+1,epoches,(i+1),loss.item()))实践任务3-手写数字识别网络训练

模型训练调用train(l)训练一轮的结果如下所示:Epoch[1/1-100],Loss:0.9250Epoch[1/1-200],Loss:0.7030Epoch[1/1-300],Loss:0.4063Epoch[1/1-400],Loss:0.3836Epoch[1/1-500],Loss:0.4059Epoch[1/1-600],Loss:0.4280Epoch[1/1-700],Loss:0.3695Epoch[1/1-800],Loss:0.3918Epoch[1/1-900],Loss:0.5331实践任务3-手写数字识别网络训练模型训练模型训练的技巧:1.数据预处理2.超参数调整3.模

温馨提示

  • 1. 本站所有资源如无特殊说明,都需要本地电脑安装OFFICE2007和PDF阅读器。图纸软件为CAD,CAXA,PROE,UG,SolidWorks等.压缩文件请下载最新的WinRAR软件解压。
  • 2. 本站的文档不包含任何第三方提供的附件图纸等,如果需要附件,请联系上传者。文件的所有权益归上传用户所有。
  • 3. 本站RAR压缩包中若带图纸,网页内容里面会有图纸预览,若没有图纸预览就没有图纸。
  • 4. 未经权益所有人同意不得将文件中的内容挪作商业或盈利用途。
  • 5. 人人文库网仅提供信息存储空间,仅对用户上传内容的表现方式做保护处理,对用户上传分享的文档内容本身不做任何修改或编辑,并不能对任何下载内容负责。
  • 6. 下载文件中如有侵权或不适当内容,请与我们联系,我们立即纠正。
  • 7. 本站不保证下载资源的准确性、安全性和完整性, 同时也不承担用户因使用这些下载资源对自己和他人造成任何形式的伤害或损失。

评论

0/150

提交评论