conda
-
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으로 해주면 해결된다...
-
[PyTorch] How to run pytorch using conda by linking vscode and jupyter notebook on OSX(mac).AI 2020. 4. 17. 14:26
pytorch로 deeplearning 예제를 로컬(osx)에서 실습해보려고 한다. 에디터는 pycharm을 많이 쓰는거 같던데, 나의 경우 항상 쓰던 vscode를 사용하고자 한다. 예제 소스는 .ipynb 파일로 되어있어서, .py 파일로 변형하지 않고 사용하기 위해 jupyter notebook을 사용하려 한다. 자 그럼 이제 개발 환경을 구축해보자. 1. Install python 3.x 링크 기본적으로 python 3.x를 설치해준다. (설치 과정은 생략..) 2. Install vscode & extensions. VSCODE 그리고 확장 프로그램을 설치한다. 설치가 완료되면 vscode를 재시작 해준다. 3. Install jupyter $ pip3 install --upgrade pip $..