这次的课程使用MNIST手写数字数据集来训练生成式对抗网络,我们可以使用该网络模拟生成手写数字图片
首选使用Mindspore的数据处理api来进行数据集处理

通过create_dict_iterator函数将数据转换成字典迭代器,然后使用matplotlib模块可视化部分训练数据,之后就能看到训练的详情

课程在判别器和生成器中采用全连接网络架构和 ReLU 激活函数构建网络架构
在训练的过程中,分别获取训练过程中的损失,并在每轮迭代结束时进行测试,将隐码批量推送到生成器中,以直观地跟踪生成器 Generator 的训练效果

最后加载生成器网络的模型生成的图片

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐