减少数据集大小时的can‘t set attribute问题

最近在学习pytorch时，想在本地跑一些代码，但是速度太慢了，于是便根据教程更换了数据集大小，代码如下：

train_set.train_data = train_set.train_data[0:5000]
train_set.train_labels = train_set.train_labels[0:5000]

test_set.test_data = test_set.test_data[0:5000]
test_set.test_labels = test_set.test_labels[0:5000]

可是运行报错了，报错结果如下：

UserWarning: train_data has been renamed data
  warnings.warn("train_data has been renamed data")
AttributeError: can't set attribute

它的提示说train_data现在重命名为data，遂更改如下：

train_set.data = train_set.data[0:5000]
train_set.train_labels = train_set.train_labels[0:5000]

test_set.data = test_set.data[0:5000]
test_set.test_labels = test_set.test_labels[0:5000]

可是代码运行依旧报错，报错结果如下：

UserWarning: train_labels has been renamed targets
  warnings.warn("train_labels has been renamed targets")
AttributeError: can't set attribute

与上个错误一样，提示说train_labels现在是targets，改之如下：

train_set.data = train_set.data[0:5000]
train_set.targets = train_set.targets[0:5000]

test_set.data = test_set.data[0:5000]
test_set.targets = test_set.targets[0:5000]

这时便运行成功，没有报错，问题解决。