我是 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) 的张量。我错过了什么?这是因为转换不是直接应用于数据加载器,而是仅在运行时应用吗?否则我如何探索我的数据?
添加回答
举报
0/150
提交
取消