num_workers
-
iter(dataloader) 시 TypeError: cannot pickle 'Environment' objectAI 2020. 12. 14. 15:20
아래의 코드와 같이 리스트에 데이터로더를 iterator로 넣고자하는데 에러가 발생 했다. _data_loader = torch.utils.data.DataLoader( _dataset, batch_size=_batch_size, shuffle=True, num_workers=int(opt.workers), collate_fn=_AlignCollate, pin_memory=True) self.dataloader_iter_list.append(iter(_data_loader)) TypeError: cannot pickle 'Environment' object append에 아무런 문제가 없어보이지는데 원인은 데이터로더 생성시 num_workers 값 떄문이다. num_workers=0으로 해주면 해결된다...