news 2026/4/17 20:31:57

day38

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
day38

@浙大疏锦行

pytorch简单神经网络的流程

1.数据预处理(归一化、转换成张量)
作用:让数据符合模型输入要求(数值范围稳定、格式为张量)。
归一化:将数据缩放到[0,1]或[-1,1]区间,避免数值差异大导致模型训练不稳定;
张量转换:PyTorch 模型仅接受张量(Tensor)作为输入,需将 Python 列表 / NumPy 数组转成张量。

示例代码:

2.模型的定义
PyTorch 中通过继承nn.Module类来定义模型,需包含 “定义层” 和 “定义前向传播” 两个核心步骤。
i. 继承nn.Module类nn.Module是 PyTorch 中所有模型 / 层的基类,提供了参数管理、设备迁移等功能。
ii. 定义每一个层在__init__方法中定义模型的层(如全连接层nn.Linear、激活函数nn.ReLU等)。
iii. 定义前向传播流程在forward方法中指定数据在层之间的流动顺序(即模型的计算逻辑)。

示例代码:

3.定义损失函数和优化器
损失函数:衡量模型预测值与真实值的差距(如回归任务用MSELoss,分类任务用CrossEntropyLoss);
优化器:根据损失调整模型参数,最小化损失(常用Adam、SGD)。

示例代码:

4.定义训练流程
核心逻辑:前向传播(算预测)→ 算损失 → 反向传播(算梯度)→ 优化器更新参数,重复多轮(epoch)。

5.可视化 loss 过程
用matplotlib绘制训练过程中损失的变化,观察模型是否收敛。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/17 16:28:30

小白指南:识别影响Intel平台USB3.0传输速度的因素

为什么你的USB3.0跑不满5Gbps?一文讲透Intel平台性能瓶颈真相 你有没有遇到过这种情况:买了一个标称读写速度500MB/s的高速U盘,插在电脑上拷贝文件,结果实测只有100多MB/s,甚至更低?明明接口是蓝色的、系统…

作者头像 李华
网站建设 2026/4/17 16:35:38

day34 GPU训练与call方法

GPU训练 要让模型在 GPU 上训练,主要是将模型和数据迁移到 GPU 设备上。 在 PyTorch 里,.to(device) 方法的作用是把张量或者模型转移到指定的计算设备(像 CPU 或者 GPU)上。 对于张量(Tensor)&#xff…

作者头像 李华
网站建设 2026/4/18 8:06:13

QListView与右键菜单集成的项目实战

如何让 QListView 的右键菜单“聪明”起来?一次实战级深度拆解在做 Qt 桌面应用时,你有没有遇到过这样的场景:用户对着一个设备列表点了右键——结果菜单弹出来全是灰的,或者干脆没反应;又或者,点了“删除”…

作者头像 李华
网站建设 2026/4/16 15:51:03

一文说清freemodbus在工控网络中的角色定位

freemodbus:嵌入式工控通信的“隐形引擎”是如何工作的?在一条自动化生产线上,PLC 正在读取十几个传感器的温度数据,HMI 屏幕实时刷新着设备状态,而远在控制室的工程师通过 SCADA 系统远程调整参数——这些看似平常的操…

作者头像 李华
网站建设 2026/4/17 12:08:11

眼睛也会“偷懒”?调节力不足,小心近视加深、视疲劳

在数字化时代,人们日均近距离用眼时长大幅增加,不少人都有过这样的体验:长时间看书、看电子屏幕后,会出现眼睛酸胀、干涩、视物模糊等不适,这其实是眼睛在发出“预警信号”——可能是调节力不足在作祟。很多人误以为视…

作者头像 李华
网站建设 2026/4/17 20:08:59

Zynq MPSoC中VDMA与GPU协同处理核心要点

VDMA与GPU如何在Zynq MPSoC上“无缝共舞”?揭秘高效图像流水线的设计精髓你有没有遇到过这样的场景:摄像头采集的1080p视频流刚进系统,还没开始处理就卡顿了;或者CPU满载跑图像算法,结果连个UI都响应不过来&#xff1f…

作者头像 李华