Pytorch训练LeNet模型MNIST数据集

地上悬河

已于 2024-06-04 16:13:05 修改

阅读量432

点赞数 3

文章标签： pytorch 人工智能 python

于 2024-05-29 11:55:53 首次发布

本文链接：https://blog.youkuaiyun.com/DUTwangtaiyu/article/details/139289960

版权

如何用torch框架训练深度学习模型（详解）

0. 需要的包

import torch
from torch.nn import CrossEntropyLoss
from torch.optim import SGD
from torch.utils.data import DataLoader
from torchvision import datasets, transforms

1. 数据加载和导入

以MNIST数据集为例

# 1.1 需要设置数据归一化
train_transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,))])
test_transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,))])
# 1.2 用dataset.MNIST函数下载和加载训练集与测试集 
train_dataset = datasets