TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found

原因:DataLoader中出现了NoneType对象。这通常意味数据集(通常是一个继承自torch.utils.data.Dataset的类)在某个时候返回了None

解决方案:

(已解决)在多进程加载数据时遇到这个问题:将DataLoadernum_workers参数设为0

data_loader = DataLoader(my_dataset, batch_size=32, shuffle=True, num_workers=0)

你可能感兴趣的:(深度学习报错,numpy)