[일반] 딥러닝 OOM질문있어요

[익명(220.81)]

3D unet 이제 시작하는 학생인데 사용중인 피시가 3090 24g인데도 oom이 발생하네요

담당자한테 지금 pc사양이 부족하다고 보고올리기전에 '이러이러해서 메모리가 부족하니 그래픽카드를 증설이 필요하다' 라고 정리해서 가고싶은데

내가 모델은 잘못 만들어서 oom이 걸리는지 아니면 연구 과제가 3090으로도 부족한 건지 증명할수있는 방법이있을까요...

텐서플로우 사용중이고 input 사이즈 줄이고 배치사이즈도 줄이고 필터맵 갯수 줄이니까 되긴한데 학습률이 많이 낮아요

[익명(220.81)]

단순히 모델 파라미터갯수로 계산할수있을까요

[익명(23.119)]

배차값, 파라메터 높일 수록 정확도가 높아지는지는 확인해봤음? 확인해서 효율이 낮으면 메모리 올린다고 될게 아님. 새로 장비 사는건 가장 최후의 수단으로 생각해라. 나중에 졸업하고 취업해서 회사에서도 마찬가지임.

[익명(223.39)]

파라미터 높여보고 싶은데 OOM 걸려서 막혀요...

[1(118.235)]

배치사이즈 줄이고 gradient accumulation step 검색하셔서 적용해보세요