使用pytorch中的transform提示 'ToTensor' object is not iterable

刚刚在Pytorch中对数据进行处理的时候，我写了下面这一行代码：

 train_folder_set = ImageFolder(filename + '/train', transform=transforms.Compose(
     transforms.ToTensor(),
    ))

然后使用一个循环输出：

for i in train_folder_set:
	pint(i)

运行，然后就报错了：

TypeError: 'ToTensor' object is not iterable

按照道理来说，不应该啊，后来才发现，是transform.Compose()使用的时候要加[]，然后将代码改成下面这张：

    train_folder_set = ImageFolder(filename + '/train', transform=transforms.Compose([
     transforms.ToTensor(),
    ]))

问题就解决了，所以现在记录一下，为可能遇到相同问题的同学提供思路。

2020 4.21