博主分享了在ModelArts平台上使用预置算法进行模型训练的经历，提到了提供GPU和Ascend910两种训练版本。博主选择了Ascend910，称赞其算力强大，9.5小时完成了1000个epochs的训练。后续进行了模型转换，使用MindStudio将.pb模型转为.om，以便在AIPP上进行颜色转换和归一化操作。
今天接着说说模型训练的部分，这部分比较简单，在ModelArts上官方提供了算法和强大的算力，这极大简化了AI开发中最为复杂的模型训练部分。这里要说明的是，这个算法提供了预置算法的版本(1种，GPU训练)和AI市场的版本(两种，GPU和Ascend 910训练)，共计三种，本质无区别，对后续部署无影响，这个可以自行选择。如果你想用GPU，可以选择GPU版本，官方也提供了Ascend 910训练的版本，我使用的是Ascend 910的版本，不得不说，Ascend 910的算力强大呀，体验很刺激。
关于如何训练这部分，毛老师在帖子中也有介绍，按照说明操作即可，基于ModelArts一站式AI开发平台，非常简单的，点击几下就行了，链接：https://bbs.huaweicloud.com/blogs/163392
我使用的是1 * Ascend 910的规格，max_epochs = 1000:
用了约9.5个小时训练完成，还不错。
之后是模型转换，这里我是将训练得到的.pb模型在自己的MindStudio中转为.om的，因为我想在AIPP上做BGR到RGB的色域转换，云端做转化的话，需要设置配置文件，比较麻烦，我就在线下转化了，同时做了每个像素除以255的归一化操作，剩下的没什么了，关于输入节点和输入尺寸可参考算法界面的介绍：
未完待续......
1835
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
