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

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

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

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

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐