使用MindSpore进行模型训练的准备工作和步骤

新知榜官方账号

2023-09-25 00:06:24

准备工作

使用MindSpore进行模型训练需要做以下的准备工作：

训练的网络模型
训练数据以及数据处理
训练回调函数
启动训练的python脚本

网络模型可以参考MindSpore的官网，根据自己的需求编写；数据处理需要考虑数据格式、数据分割、数据增强等问题，可以使用MindSpore提供的transformations模块进行处理；训练回调函数可以用于查看模型的精度、保存模型以及输出日志信息；启动训练的python脚本需要进行参数解析，同时需要在云端NPU中进行启动。

数据处理

对于分类模型，数据格式是一张图片对应一个标签。可以使用一个类将硬盘数据转换为MindSpore模型，也可以使用列表、迭代器，其中比较重要的是__len__以及__getitem__这两个方法，作用是计算数据个数以及使用索引的方式获取数据。

数据分割可以手工分割，也可以通过代码实现。可以使用列表保存几个分割后的子集。数据转换可以使用MindSpore的dataset.GeneratorDataset将数据转换为MindSpore的数据格式，需要考虑图片大小放缩、随机水平翻转、随机角度旋转、数据类型转换以及将(h,w,c)通道的格式转换为(c,h,w)格式等问题。

训练回调

训练时如果需要查看模型的精度，以及保存模型训练过程中迭代最优的结果，需要重载mindspore.train.callback.Callback类。可以定义一个AccuracyMonitor类用于监控训练精度，可以定义get_accuracy和get_loss方法用于计算精度和损失函数，同时在epoch_end方法中保存最优模型。

云端NPU

使用云端NPU进行训练，需要进行以下工作：

建立文件夹
上传数据
上传算法
创建算法
创建训练

具体的步骤可以参考华为云的Modelarts界面，需要设置输入文件路径、输出文件路径、超参数、日志等信息。

本页网址：https://www.xinzhibang.net/article_detail-13475.html

寻求报道，请

关键词

MindSpore 模型训练数据处理训练回调云端NPU

分享至微信：

扎心文案+AI插画=爆款！揭秘8万赞视频的制作全流程

又一个爆款诞生！今天拆解一个账号——@尴尬的老男人，内容很简单，最高一条近8万赞，但平均每条视频都有几百赞！他的视频看似就是几张图片配上一段音乐，却精准狙击用户对“生活氛围感”和“情绪价值”的深层渴望，再加上温暖色调、治愈配乐——每一帧都让人忍不住驻足回味。其实，这种视频的关键就是2个：扎心文案

分类标签

08-12 10:08

三个国外专业免费的调色网站

三个国外专业免费的调色网站无论制作视频、编辑图片，合适的色彩会让你的作品增加视觉冲击力，所以专业的调色技术价值非凡，一部电影或一部好的视频、图片海报，专业调色耗费少则上万，多则上百万，只是大家对专业调色这个行业知之甚少，今天网玩君就为大家分享三个国外专业免费的调色网站，记得收藏好，以后编辑视频、图片

分类标签调色网站免费

04-18 16:42

第六届中国国际工业设计博览会盛况空前

第六届中国国际工业设计博览会于11月30日至12月2日在武汉举行。本届博览会展览面积达2万平方米，参展企业超过300家，其中，参展国家级工业设计中心数量为历届最多。众多国内外行业龙头企业展示了多项“首台套”的硬核实力，为中国制造业向高端化、智能化、绿色化转型升级赋能。中铁宝桥集团展示了多个“国内首条

分类标签中国国际工业设计博览会工业设计创新成果中国制造业

04-18 16:42

知识分享查理芒格思考的独立性与决策的准确性

知识分享查理芒格思考的独立性与决策的准确性本文介绍了成功投资者必须具备独立思考和决策准确性，同时保持谦逊低调的心态，以及成功的关键因素之一——运气。同时，本文还分享了下周市场关注的行业和概念。一、知识分享查理芒格思考的独立性成功的投资者必须具备独立思考的能力，因为只有独立思考才能避免盲目跟风和投资决

分类标签独立思考决策准确性谦逊低调

04-18 16:42

阿里国际发布三款AI设计工具助力中国制造走向全球

阿里国际发布三款AI设计工具助力中国制造走向全球12月1日，在第六届中国国际工业设计博览会上，阿里国际数字商业集团（简称“阿里国际”）发布了3款设计生态工具：堆友、PicCopilot、鹿班AI。这3款产品具有AI绘画、AI模型创作、AI图像和视频处理等功能，用AI设计服务全球商家更轻松的做好本地市

分类标签 AI设计阿里国际数字商业集团

04-18 16:42

ChatGPT Midjourney 文心一言文心一格 bing新必应 Stable diffusion 稿定设计墨刀AI mastergo Adobe Firefly runway