TypeError: Caught TypeError in DataLoader worker process 0. TypeError: ‘NoneType‘ object is not sub_

使用pytorch 进行train 出现TypeError: Caught TypeError in DataLoader worker process 0. TypeError: 'NoneType' object is not sub_

两个错误，解决办法。

第一步：分析错误原因

我们可以看到第一个错误是关于DataLoader进行训练数据的存取工作线程的问题，

第二个问题可以看出是对象不存在，没有这个类型。也就是说函数可能读取不到数据。

第二步：解决问题

首先，我们由第二个问题可以确定一下是否是路径的问题，常常因为一些特殊符号导致路径出现问题

多进程问题可能由于内存的原因导致出现错误，可以将进程改为单进程，

dataloader = torch.utils.data.DataLoader(
        dataset,
        batch_size=opt.batch_size,
        shuffle=True,
        num_workers=opt.work,
        pin_memory=True,
        collate_fn=dataset.3dffa,
    )
将num_workers=opt.work改为num_workers=1