我是 pytorch 的新手,想了解一些东西。
我正在按如下方式加载 MNIST:
transform_train = transforms.Compose(
[transforms.ToTensor(),
transforms.Resize(size, interpolation=2),
# transforms.Grayscale(num_output_channels=1),
transforms.RandomHorizontalFlip(p=0.5),
transforms.Normalize((mean), (std))])
trainset = torchvision.datasets.MNIST(root='./data', train=True,
download=True, transform=transform_train)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=batch_size,
shuffle=True, num_workers=2)
然而,当我探索数据集时,即trainloader.dataset.train_data[0],我得到了一个范围为 [0,255] 且形状为 (28,28) 的张量。
我错过了什么?这是因为转换不是直接应用于数据加载器,而是仅在运行时应用吗?否则我如何探索我的数据?
相关分类