难度:新手入门⭐ 语言:Python3、Pytorch 🍺 要求: 学习如何编写一个完整的深度学习程序 手动推导卷积层与池化层的计算过程 文章目录 一、 前期准备 1. 设置GPU 2. 导入数据 3. 数据可视化 二、构建简单的CNN网络 三、 训练模型 1. 设置超参数 2. 编写训练函数 3. 编写测试函数 4. 正式训练 四、 结果可视化 🏡 我的环境: 语言环境:Python3.8 编译器:jupyter notebook 深度学习环境:Pytorch 一、 前期准备 1. 设置GPU 如果设备上支持GPU就使用GPU,否则使用CPU import torch import torch.nn as nn import matplotlib.pyplot as plt