使用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)...
原创
2020-04-21 22:29:33 ·
3874 阅读 ·
4 评论