标签：pytorch

推理部署时，模型能输出结果并不等于代码已经进入正确推理状态。深度学习项目若仍保留训练图、随机层或不必要的中间张量，会造成显存升高、延...

图像分类 分类器就是一个从像素到类别概率分布的函数。其他一切都是管道工程。 类型： 构建 语言： Python 前置要求： 第2阶段第09课（模型评估），第3阶...

从PyTorch导出ONNX时一个简单的reshape变成了几十个算子的“意大利面”？维度不匹配导致TensorRT编译失败？模型体积膨胀推理延迟居高不下？本文...

文章目录一、前言二、数据集说明三、环境依赖四、数据加载五、全连接神经网络（BP 神经网络）5.1 网络结构5.2 核心原理5.3 代...

前言 YOLOv8 是 Ultralytics 推出的最新一代目标检测模型，广泛应用于工业检测、智能监控、自动驾驶等领域。在实际部署时ÿ...

最近学习了一些深度学习和 PyTorch 框架的基础知识，并尝试使用 MNIST 手写数字数据集完成一个简单的图像分类任务。虽然手写数字识别是深度...

AI 的引擎：自动微分到底是怎么工作的？深度学习能训练，靠的是反向传播；反向传播的数学本质，...

系列：HarmonyOS 开发入门 · 08上一篇解决“为什么选 Navigation”，这一篇直接把一套能用于真实项目的基础结构搭...

一、model.py 完整解析model.py 是整个项目的核心，只有 330 行代码，却实现了完整的 GPT 模型。1. Laye...