PyTorch Dataloader当使用LMDB数据时num_worker>0引发TypeError: can‘t pickle Environment objects问题的解决
PyTorch Dataloader当使用LMDB数据时num_worker>0引发TypeError: can't pickle Environment objects问题的解决防忘记,原issue:https://github.com/pytorch/vision/issues/689#issuecomment-787215916解决方法:不要在__init__方法中调用lmdb.open方法;在第一次加载数据时打开lmdb。代码如下class DataLoader(torch.u
翻译
2021-08-09 14:57:11 ·
1824 阅读 ·
1 评论