昇思25天学习打卡营第二天|快速入门

今天主要的学习内容是MindSpore的快速入门,主要是跟着教程过了一遍图像分类问题的流程,主要包含数据获取和处理、网络构建、模型训练、保存模型和加载模型、预测推理五个流程,详细内容可以参考官方文档,这里只说一下我的心得。

数据获取和处理

这里主要使用download模块获取数据集,并使用MindSpore自带的mindspore.dataset.MnistDataset获取数据对象并处理,如果之前学过Pytorch或Tensorflow的应该可以很快上手这部分内容。

网络构建

这里就是使用几个全连接层和Relu激活函数搭建的一个MLP网络,其搭建流程个人感觉和Pytorch差不多,其中construct函数类似于forword函数。

模型训练

这里首先定义了损失函数、优化器、训练流程、测试流程等,如果学过神经网络相关的基础知识,总体上理解这部分的流程还是比较容易的,这里附上我的训练效果图
在这里插入图片描述

保存和加载模型

保存模型一行代码的事,没什么好说的,保存格式为.ckpt
加载模型时是先使用load_checkpoint函数导入模型,再使用load_param_into_net函数将模型参数导入到网络中,该函数返回值包括未被加载的参数列表,可以看到哪些参数未被加载,这是一个很实用的功能。

预测推理

使用加载模型的网络对测试集中的图像进行预测得到最终结果

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值