Pytorch代码实战（MINST-FASHION数据集）

原创

已于 2024-02-22 11:03:37 修改 · 545 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #人工智能 #python

于 2024-02-22 11:03:21 首次发布

import torch
from torch import nn
from torch import optim
from torch.nn import functional as F
from torch.utils.data import DataLoader, TensorDataset


import torchvision
import torchvision.transforms as tranfsorms

import matplotlib.pyplot as plt
import numpy

# 确定数据，确定超参数
lr = 0.15
gamma = 0
epochs = 10
bs = 128

# 实例化数据集
mnist = torchvision.datasets.FashionMNIST(root = "D:\jupyterDemo\MINST-FASHION数据集"
                                          ,download = False
                                          ,train = True
                                          ,transform = tranfsorms.ToTensor()
                                         )

# 数据样式探索
batchdata = DataLoader(mnist
                       ,batch_size = bs
                       ,shuffle = True
)

for x,y in batchdata:
    print(x.shape)
    print(y.sh