이미지 합성 인공지능 #3

RuntimeError: CUDA out of memory. Tried to allocate... 해결하려는 중 ..

 

 

 

...해야지 뭐

[파이토치] 오류 해결 RuntimeError: CUDA out of memory. Tried to allocate...

위 블로그의 해결방법 1) 적용해 봄

 

 

 

해결방법 3)도 적용해 봄..

train, test input tensor에 .to(torch.int64) 붙여주면 된다고 하는데

내가 붙여준 자리가 맞는진 모르겠다 'ㅅ'

 

 

 

RuntimeError: CUDA out of memory. Tried to allocate... 에러는 사라졌는데

쟤는 여전히 남아있다

 

 

 

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

입력 텐서는 GPU 상에 있는데 model weights는 아니라서 에러 뜨는 것 같다

model weights를 GPU로 보내주면 된다는데

이것 저것 시도하다가 그냥 방법을 바꿔버렸다....!

 

 

 

CPU에서 돌려주기로..

GPU에서 하는 방법은 나중에 생각하고

일단 결과 먼저 내보자!!!

Hardware accelerator를 GPU에서 None으로 해줬다

 

 

 

GPU에서 실행안해도 되니까 필요없는 부분 다시 깔끔하게 없애고

 

 

 

이 부분 실행시키고 기도했다.......☆★

 

 

 

쫘라란

코딩의 맛 ♡♥~~~

 

 

 

배경이 될 이미지 코랩에 올리고 확인

 

 

 

배경 이미지 변수 저장하고 전경에 맞게 resize 수행

 

 

 

전경과 배경 분리해주기

컬러 팔레트에서 강아지(dog)는 12번이기 때문에

mask = (pred == 12).astype(float) * 255

 

 

 

앙 너모 귀여워,...

 

 

 

거대 고양이가 떠오르고,.?

 

 

 

뀨앙 귀여워~~~~~~

여기서 끝내면 아쉬우니까 응용버전은 다음 포스트에!