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

【问题标题】：AttributeError: can't set attribute in splitting MNIST datasetAttributeError：无法在拆分 MNIST 数据集时设置属性
【发布时间】：2021-09-04 10:17:10
【问题描述】：

我正在使用 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

如何解决这个错误？

【问题讨论】：

你确定对象mnist_data中存在属性“train_data”吗？
是的，还有train_labels 属性和test_labels。当我改用data 和targets 属性时，问题似乎解决了。

标签： python pytorch torchvision

【解决方案1】：

如果您使用torchvision.dataset.MNIST，您可以将train=True/False 更改为您的训练集或测试集。

从docs 开始，MNIST 类有@property train_data，所以不能将 train_data 设置为属性。你可以改成mnist_data.data_train = ...。

【讨论】：