摘要:环境: python 3.11.10 pytorch 2.3.0 一、前期准备 1.设置GPU import torch import torch.nn as nn import matplotlib.pyplot as plt import torchvision device = torch.d
阅读全文
随笔分类 - 深度学习
摘要:环境: python 3.11.10 pytorch 2.3.0 本文通过PyTorch框架来构建、训练以及评估一个简单的全连接神经网络,以便理解神经网络的基本结构,并通过实际操作获得第一手的经验。选择的任务是在经典的MNIST手写数字数据集上进行数字识别,这是学习深度学习不可或缺的一个实验。 一、
阅读全文