电子工程世界-论坛»论坛 › 电子技术交流 › 嵌入式系统 › AI挑战营第一站#pytorch训练MNIST数据集实现

阅 3707|回 1

205 帖子	0 TA的资源

一粒金砂（高级）

楼主

发表于2024-4-25 00:21 只看该作者

AI挑战营第一站#pytorch训练MNIST数据集实现 [复制链接]

参考CPU安装pytorch流程安装环境

https://blog.csdn.net/qiaoxinyu1989/article/details/135953830

先安装Anaconda

创建虚拟环境

安装pytorch

安装python开发IDE PyCharm

配置python环境

创建工程

参考https://blog.csdn.net/miss_liangrm/article/details/136516286

实验MNIST手写数字识别模型训练

载入头文件

下载加载数据集

输出图像

构建CNN网络

设置超参数

训练函数

测试函数

主函数

训练结果

MODE_NUM.onnx (494.21 KB, 下载次数: 0)

MODE.pth (478.9 KB, 下载次数: 0)

此帖出自嵌入式系统论坛

最新回复

nian1206

标记一下，后面用到了好学习学习，非常感谢分享，好人一生平安！详情回复发表于 2024-6-6 17:58

点赞关注

嵌入式论坛嵌入式开发嵌入式技术嵌入式教程

nian1206

90 帖子	0 TA的资源

一粒金砂（中级）

沙发

发表于2024-6-6 17:58 只看该作者

标记一下，后面用到了好学习学习，非常感谢分享，好人一生平安！

此帖出自嵌入式系统论坛

嵌入式培训嵌入式系统嵌入式学习

返回列表发新帖回复

活动更多>>

开源项目更多>>

随便看看

#AI挑战营第一站#pytorch训练MNIST数据集实现手写数字识别
[i=s]本帖最后由LitchiCheng于2024-4-1822:28编辑[/i][attach]803168[/attach]下载MNIST数据集#MNIST数据集，用于训练，一次抓60sizeselfutilsDataLoader(torchvisionMNIST('data/',train=True,download=True,transform=torchvisionCompose([torchvisionToTen ...
#AI挑战营第一站#pytorch训练MNIST数据集 (补充测试集的准确率结果）
如在之前（https://bbscomthread-1279506-1-1.html）所述，采用了自建模型来进行训练。如果是模型1：[attach]806957[/attach]运行结果[attach]806958[/attach]使用模型2，在卷积层间中插入丢弃，虽然会降低精度，但性能应该有所提升[attach]806959[/attach]训 ...
请问大佬们，高频电子线路这门课如何学？，以后想做这方面工作
我之前做高频通讯类题的电赛，做了一年，感觉很有意思，虽然有些bug让人烧脑，结果到了老师上这门课的时候，老师只会讲结论，给做题，讲的很多的东西听着也不靠谱，比如说什么只需要知道结论就行，推导不需要了解，就我的电赛经历来看，指导老师经常是只给个 ...
【读书】《RISC-V 体系结构编程与实践（第2版）》-- RISC-V 架构寄存器学习
《计算机视觉之PyTorch数字图像处理》--全书概览
合众达仿真器安装指导
【2022得捷电子创新设计大赛】到货啦。开箱！
本周精彩博文分享
什么是单片机
display 驱动加载中调用文件系统中的图片显示出来