머신러닝
-
HardNet 개발딥러닝/pytorch 2020. 2. 5. 16:30
하도 정리가 안되서 정리용 블로그 시작 이미지에서 segmentation 을 추출하는 모델 input 이미지의 width,height와 output 이미지의 width, height가 동일 input은 [channel, width, height] 라면 output 은 [classes, width, height] 로 각 픽셀별로 이미지화하여 나오게 됨 논문은 아래 링크 https://arxiv.org/abs/1909.00948 HarDNet: A Low Memory Traffic Network State-of-the-art neural network architectures such as ResNet, MobileNet, and DenseNet have achieved outstanding accuracy..
-
pytorch gpu parallel 설정딥러닝/pytorch 2020. 2. 4. 17:44
언제나 그렇듯 gpu 설정은 빡치는 작업이다 nvidia-docker로 띄우고 끝나는게 아니라 cudnn 도 설정해줘야하고 등등등...... 아무튼 환경 설정은 어찌어찌 해서 gpu를 사용하는데 torch.nn.DataParallel을 사용해서 학습을 시키는데 메모리만 잡아먹고 오른쪽 사용량은 100%와 0%를 1:1 비율로 오락가락하는 모습을 보여준다 (딥빡) 그래서 아래 링크를 보고 gpu 병렬 처리를 다시 설정하려는데 https://medium.com/daangn/pytorch-multi-gpu-%ED%95%99%EC%8A%B5-%EC%A0%9C%EB%8C%80%EB%A1%9C-%ED%95%98%EA%B8%B0-27270617936b 🔥PyTorch Multi-GPU 학습 제대로 하기 PyTorch..