本部分介绍一些PyTorch中常见的报错信息及其解决方法
报错1:
ValueError: num_samples should be a positive integer value, but got num_samples=0
可能的原因:传入的Dataset中的len(self.data_info)==0,即传入该dataloader的dataset里没有数据
解决方法:
- 检查dataset中的路径,路径不对,读取不到数据。
- 检查Dataset的__len__()函数为何输出为零
报错2:
TypeError: pic should be PIL Image or ndarray. Got <class 'torch.Tensor'>
可能的原因:当前操作需要PIL Image或ndarray数据类型,但传入了Tensor
解决
- 检查transform中是否存在两次ToTensor()方法
- 检查transform中每一个操作的数据类型变化
报错3:
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 93 and 89 in dimension 1 at /Users/soumith/code/builder/wheel/pytorch