原因:DataLoader中出现了NoneType
对象。这通常意味数据集(通常是一个继承自torch.utils.data.Dataset
的类)在某个时候返回了None
。
解决方案:
(已解决)在多进程加载数据时遇到这个问题:将DataLoader
的num_workers
参数设为0
data_loader = DataLoader(my_dataset, batch_size=32, shuffle=True, num_workers=0)
原因:DataLoader中出现了NoneType
对象。这通常意味数据集(通常是一个继承自torch.utils.data.Dataset
的类)在某个时候返回了None
。
解决方案:
(已解决)在多进程加载数据时遇到这个问题:将DataLoader
的num_workers
参数设为0
data_loader = DataLoader(my_dataset, batch_size=32, shuffle=True, num_workers=0)