Pytorch学习笔记【10】：实战！CNN手写数字识别

最新推荐文章于 2024-08-07 13:23:50 发布

原创

最新推荐文章于 2024-08-07 13:23:50 发布 · 381 阅读

1 ·

CC 4.0 BY-SA版权

注意看代码注释！！！解析都在注释里，不要怕麻烦，自己复制代码，运行一遍，然后结合运行结果和代码注释就能够看懂代码

1. 代码：

import os

# 引入一些包
import torch
import torch.nn as nn
import torch.utils.data as Data
import torchvision
import matplotlib.pyplot as plt


# 定义一些参数
EPOCH = 1               # 训练数据的次数，我们这里假定训练一次
BATCH_SIZE = 50         # 每次训练的数据量，这个会产生每一次训练分多少次进行，或者多少批进行
LR = 0.001              # 学习率
DOWNLOAD_MNIST = False


# 下载并且加载数据集
if not(os.path.exists('./mnist/')) or not os.listdir('./mnist/'):
    # not mnist dir or mnist is empyt dir
    DOWNLOAD_MNIST = True

train_data = torchvision.datasets.MNIST(
    root='./mnist/',
    train=True,                                     # 表示训练数据
    transform=torchvision.transforms.ToTensor(),    # 将数据转换成tensor
                                                    # torch.FloatTensor of shape (C x H x W) and normalize in the range [0.0, 1.0]
    download=DOWNLOAD_MNIST,
)

# 画出一个例子，更直观
print(train_data.train_data