Pay it Forward

Pytorch를 사용하여 모델을 돌리던 중, augmentation으로써 RandomRotation을 사용하고, 이미지가 돌아갔을 때(rotate) 생기는 여백의 공간을 0으로 채우기 위하여 아래와 같이 코드를 작성하였다. from torchvision import transforms as transforms trans = transforms.RandomRotation(degrees=args.aug_degree, fill = (0,)) img_trans = trans(img) 그러나 다음과 같은 에러를 맞이하였다. UserWarning: Argument fill/fillcolor is not supported for Tensor input 알아본 결과 해결방법은 아래와 같다. Torchvision ver..

Neural network를 train을 하다보면 성능을 비교하기 위해 loss function, optimizer 등이 상이한 여러 가지 model을 돌려볼 일이 수도 없이 많다. 이 때, 한 장의 GPU만 있다면 어쩔 수 없지만 여러장의 GPU가 존재한다면 A model은 GPU 0번에서, B model은 GPU 1번에서 동시에 두 개의 model을 돌려 빠르게 결과를 보고 싶을 것이다. 이를 위하여 각 model을 특정 GPU에서 돌리는 법을 알아보고자 한다. 추가적으로, network의 weight param이 크거나 input dataset의 크기가 커서 하나의 GPU memory로는 model을 돌리기에 부족하여 CUDA out of memory error가 뜨는 경우가 많다. 이를 해결하기 ..

Pytorch-torchvision의 transforms 라이브러리를 사용하여 Data augmentation을 진행하던 도중 마주친 에러. TypeError: function takes exactly 1 argument (3 given) Stack overflow에 따르면 torch와 Pillow version의 호환문제로 인해 발생한 것으로, 다음과 같이 fill = (0,) 을 RandomRotation 함수에 추가하여 해결할 수 있다. aug_methods = transforms.RandomRotation(30, expand=False, fill=(0,)) 아래는 코드 전문.