卷积神经网络的快速入门(pytorch)

「已注销」

于 2022-08-20 15:04:31 发布

阅读量265

点赞数 1

文章标签：深度学习神经网络 pytorch python

本文链接：https://blog.csdn.net/m0_73089087/article/details/126440057

版权

卷积神经网络的快速入门(pytorch)

摘要由CSDN通过智能技术生成

import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor


# Download training data from open datasets.
training_data = datasets.FashionMNIST(
    root="data",
    train=True,
    download=True,
    transform=ToTensor(),
)

# Download test data from open datasets.
test_data = datasets.FashionMNIST(
    root="data",
    train=False,
    download=True,
    transform=ToTensor(),
)



batch_size = 64

# Create data loaders.
train_dataloader = DataLoader(training_data, batch_size=batch_size)
test_dataloader = DataLoader(test_data, batch_size=batch_size)

for X, y in test_dataloader:
    print(f"Shape of X [N, C, H, W]: {X.shape}")
    print(f"Shape of y: {y.shape} {y.dtype}")
    break




# Get cpu or gpu device for training.
device = "cuda" if torch.cuda.is_av