属性错误：无法在拆分 MNIST 数据集时设置属性

Question

我使用pytorchtorchvision.datasets.MNIST

加载我使用的数据集：

mnist_data = datasets.MNIST('../data', train=True, download=True,
transform=transforms.Compose(
[transforms.ToTensor(),transforms.Normalize((0.1307,), (0.3081,))]))

并将数据拆分为我使用的训练数据集：

mnist_data.train_data = (mnist_data.train_data.type(torch.FloatTensor)/255).bernoulli()

我得到错误：AttributeError: can't set attribute

如何解决此错误？

Answer 1

如果使用torchvision.dataset.MNIST，则可以更改列车或测试集的train=True/False。

从文档来看，MNIST类具有@propertytrain_data，因此不能将train_data设置为属性。您可以将其更改为mnist_data.data_train = ...。