[일반] 텐서플로우 compile 이후 GPU Memory문제 질문여

[익명(59.9)]

Sequencial model을 쌓고 compile하면 레이어가 한개든 두개든 세개든 무잭건 gpu메모리 1기가를 먹어버리는데 어찌 해결하나여?

[익명(118.235)]

배치 사이즈 줄여보셔요. VGG19의 파라미터도 1기가보다 한참 작았는데 모델이 1기가라는건 데이터 미니배치가 문제 아닐까요. 확실하진 않습니다...

[익명(59.9)]

감사합니다..