[error] CUDA out of memory

ivor·2022년 5월 2일
0

problems

목록 보기
3/4

서버에 모델을 올리고 학습하려던 중 낯선 에러가 떴다.

메모리가 부족하다길래 확인해봤다.

음? 전혀 부족한 것 없는데..?

열심히 구글링을 해봤지만 해결되는 것은 없었다. 그러던 중 문득 든 생각 하나.

당시 이미지 분류 대회를 하며 베이스라인 코드를 변경하고 있었는데 이미지 전처리 과정에서 내가 resize를 크게 늘렸었다. 당연히 파라미터는 많아지고 요구 메모리 또한 많아졌던 것이다.
'batch size는 똑같은데..?'하며 다른 이유를 찾고 있었는데 생각보다 간단한 이유였다.

resize에 맞게 메모리 초과가 발생하지 않는 적절한 batch size를 찾아서 실험을 재개할 수 있었다.

profile
BEST? BETTER!

0개의 댓글