TypeError: can't pickle _thread._local objects

can’t pickle _thread._local objects

在调试一个pytorch的image-caption代码时，出现了这个报错

报错的意思应该是_thread._local类型无法用pickle操作。向上看可以发现好像问题和多进程的进程池有关系。
问题应该就发生在train_loader上，它就是一个普通的DataLoader实例

train_loader = torch.utils.data.DataLoader(
    CaptionDataset_tra(data_folder, data_name, 'TRAIN', 
    transform=transforms.Compose([normalize])),
    batch_size=batch_size, shuffle=True, num_workers=workers, pin_memory=True)
    

这里CaptionDataset_tr读入的应该是一个h5py格式的文件。尝试了很多方法都没能改好。最后不得不把num_workers改成了0(原本workers=1)，强制不使用多进程。问题解决。
看其他博客有说是因为不能在多进程中使用自己定义的类型，或者是和字典有关的问题，确实没看懂。

代码来源https://github.com/tqxg2018/Image-Caption-Pytorch
其中的train-traditional.py