发新帖

关于python“AttributeError: can't set attribute”的问题

[复制链接]
270 1
代码:
class PermutedMNIST(datasets.MNIST):

    def __init__(self, root="~/.torch/data/mnist", train=True, permute_idx=None):
        super(PermutedMNIST, self).__init__(root, train, download=True)
        assert len(permute_idx) == 28 * 28
        if self.train:
            self.train_data = torch.stack([img.float().view(-1)[permute_idx] / 255
                                           for img in self.train_data])
        else:
            self.test_data = torch.stack([img.float().view(-1)[permute_idx] / 255
                                          for img in self.test_data])

错误:
AttributeError                            Traceback (most recent call last)
<ipython-input-12-dd28c73426fa> in <module>
     13
     14
---> 15 train_loader, test_loader = get_permute_mnist()

<ipython-input-12-dd28c73426fa> in get_permute_mnist()
      4     idx = list(range(28 * 28))
      5     for i in range(num_task):
----> 6         train_loader = torch.utils.data.DataLoader(PermutedMNIST(train=True, permute_idx=idx),
      7                                                       batch_size=batch_size,
      8                                                       num_workers=4)

~/syk/code/ewc.pytorch-master/data.py in __init__(self, root, train, permute_idx)
     11         if self.train:
     12             self.train_data = torch.stack([img.float().view(-1)[permute_idx] / 255
---> 13                                            for img in self.train_data])
     14         else:
     15             self.test_data = torch.stack([img.float().view(-1)[permute_idx] / 255

AttributeError: can't set attribute


我知道答案 回答被采纳将会获得10 金币 + 5 金币 已有1人回答

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
本楼点评(0) 收起

精彩评论1

HsinJhao  TF荚荚  发表于 2019-12-10 13:58:51 | 显示全部楼层
这应该是用的pytorch框架来进行数据处理吧?
我个人不会pytorch,无法解答你的问题,抱歉。
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

快速回复 返回顶部 返回列表