news 2026/6/10 17:45:40

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/6/10 11:05:30

GitHub热门项目都在用的PyTorch环境配置方式揭秘

PyTorch-CUDA镜像:现代深度学习项目的环境基石 在AI项目开发中,最令人沮丧的场景之一莫过于——代码写完了,模型设计好了,结果运行时却报出 torch.cuda.is_available() 返回 False。更糟的是,同事跑得好好的训练脚本&a…

作者头像 李华
网站建设 2026/6/5 12:57:14

Day41 图像数据与显存

浙大疏锦行 import pandas as pd import numpy as np import matplotlib.pyplot as plt import seaborn as sns import warnings import time from sklearn.model_selection import train_test_split from sklearn.metrics import accuracy_score, precision_score, recall_sc…

作者头像 李华
网站建设 2026/5/31 5:20:55

PyTorch-CUDA-v2.6镜像中的CUDA工具包包含哪些核心组件?

PyTorch-CUDA-v2.6镜像中的CUDA工具包包含哪些核心组件? 在深度学习工程实践中,一个常见的痛点是:明明代码写得没问题,模型结构也合理,可一运行就报错——不是CUDA不可用,就是cuDNN加载失败。更糟的是&…

作者头像 李华
网站建设 2026/6/10 15:08:11

PyTorch-CUDA-v2.6镜像如何调用多块GPU进行并行计算

PyTorch-CUDA-v2.6镜像如何调用多块GPU进行并行计算 在现代深度学习项目中,模型复杂度与数据规模的指数级增长使得单卡训练逐渐成为性能瓶颈。哪怕是最新的RTX 4090,在面对百亿参数的大模型时也显得力不从心。这时候,我们真正需要的不是更强的…

作者头像 李华
网站建设 2026/5/24 0:45:57

PyTorch-CUDA-v2.6镜像如何帮助你节省至少80%环境配置时间

PyTorch-CUDA-v2.6镜像如何帮助你节省至少80%环境配置时间 在深度学习项目启动的前48小时里,有多少开发者真正把时间花在了模型设计上?更多时候,我们都在和驱动版本、CUDA兼容性、cuDNN缺失这些“老朋友”打交道。明明买的是A100服务器&…

作者头像 李华
网站建设 2026/6/10 15:05:55

Day42 图像数据与显存

# 先继续之前的代码 import torch import torch.nn as nn import torch.optim as optim from torch.utils.data import DataLoader , Dataset # DataLoader 是 PyTorch 中用于加载数据的工具 from torchvision import datasets, transforms # torchvision 是一个用于计算机视觉…

作者头像 李华