知识库
PyTorch内置数据集简介
2025-05-01 22:20:22 李腾 1 次阅读
内置数据集
示例代码
from torchvision import datasets, transforms
transform = transforms.Compose(
[transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))]
)
train_dataset = datasets.MNIST(
root="./data", train=True, transform=transform, download=True
)
test_dataset = datasets.MNIST(
root="./data", train=False, transform=transform, download=True
)
print(train_dataset)
print(test_dataset)
转载请注明转自www.hylab.cn，原文地址：PyTorch内置数据集简介