인프런 커뮤니티 질문&답변
작성한 질문수
training
작성
·
270
1
안녕하세요 수업 잘 듣고 있습니다. 제 input 파일들 (이미지, annotated file)로 Vision transformer를 colab에서 훈련해보려고 시도해보았습니다. 그런데, RuntimeError:CUDA out of memory가 뜨더라구요. 제 질문은
cifar10과 수업시간에 사용하신 hyperparameters로 colab에서 훈련을 해보셨는지요?
CUDA out of memory를 피할 수 있는 방법이 없을까요? 제 input image가 720x496 grayscale일인 관계로 batch_size=1, patch_size=4로 설정을 해보았지만, colab에서 주는 GPU사이즈로는 감당하기가 어렵네요.
답변 1
0
딥러닝호형
지식공유자
안녕하세요.
코랩에서 구동이 가능하도록 CIFAR10(32x32) 사용과 가동 가능한 하이퍼 파라메터들을 설정하고 실제 학습 및 모든 실행을 코랩에서 진행했습니다. 파일 그대로를 실행하시면 작동이 될 것입니다.
현재 이미지 사이즈를 720x496을 사용하셨다는 것인가요? 이 정도 크기면 코랩에서는 GPU 메모리 부족 현상을 해결하기 어렵습니다.
고성능 GPU 환경은 딥러닝을 배우는데 큰 장애물 중 하나죠 ㅠㅠ
감사합니다.