이미지 기반 질의 응답 AI 경진대회 월간 데이콘

다들 실행환경이 어떻게 되시나요?

2023.07.18 17:09 1,431 Views

저는 RTX 3050 4GB 하나로 수행해보려고 하는데, 자꾸 outofmemory가 생겨서 혹시 다른 분들은 어떤 환경에서 학습과 추론을 진행하는지 궁금합니다.


자원은 충분한데 제가 코드를 잘못 짜서 그런건지, 아니면 원래 부족해서 다른 방법을 강구해봐야하는지 고민입니다. 조언주시면 감사하겠습니다.

로그인이 필요합니다
0 / 1000
flyinghooniman
2023.07.18 17:25

colab으로 test결과 베이스라인 기준으로 32 batch로 돌렸을 때 9.9GB 정도 리소스를 잡아먹네요! 자문자답 죄송합니다.

NN_is_all_you_need
2023.07.18 17:50

그럼 한 Batch size 4 정도로 하시면 로컬 머신으로도 돌아가지않을까요?

flyinghooniman
2023.07.18 23:19

로컬에서는 이상하게 batch를 1로 줄여도 오류가 나네요 ㅠㅠ 그래서 일단  colab으로 사용중입니다 

EISLab_이희원
2023.07.18 19:58

저는 RTX 3090Ti 2장의 환경에서 실행합니다.

flyinghooniman
2023.07.18 23:19

알려주셔서 감사합니다

귀여미
2023.07.25 18:43

무료 colab에서 batch size 64로 실행중입니다.
kaggle에서 t4 그래픽카드 2개를 이용할 수 있으니 참고하세요