pytorch 相关问题汇总

版权声明：任意转载 https://blog.csdn.net/Z609834342/article/details/83858070

问题1：不同的数据集下面label数组的名字不同。

svhn: train_dataset.labels
其他torchvision自带的为: train_dataset.train_labels

问题2：因为需要所以在如datasets.MNIST下载数据集和生成DataLoader中间进行数据处理，发现在都使用ToTensorheNormalize的情况下MNIST生成的data和label是tensor，而CIFAR10和SVHN却是数组型，而且没有进行相应的归一化。

问题3：进行RandomCrop操作时，images要是PIL image类型。

问题4：验证集的提取https://am207.github.io/2018spring/wiki/ValidationSplits.html