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

Question

我正在使用 pytorch torchvision.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。

从 docs 开始，MNIST 类具有 @property train_data，因此您无法将 train_data 设置为属性。您可以将其更改为 mnist_data.data_train = ...。