안녕하세요.&nbsp;
강의 잘 듣고 있습니다. 머신러닝 완벽 가이드에서 CNN fundamental 편까지 잘 들었습니다.
강의 내용을 바탕으로 plant image classfication 에 cross validation을 적용해보려고,&nbsp;
돌려보면 3번째 폴드를 어느 정도 돌고 나서는 OOM 이슈가 발생하는데 메모리 부족 현상을 해결할 수 있는 방법이 있을까요?
개인 워크스테이션을 사용하는데 cross validation을 적용하지 않으면 원활히 돌아가는데 KFold를 적용해 5 폴드 설정 후&nbsp;
시행하면 OOM 이슈가 발생하네요.
좋은 답변 혹은 팁 부탁 드리겠습니다. 감사합니다.

안녕하십니까,&nbsp;
일반적으로는 한번 수행할 때 문제가 없으면, 메모리를 재 활용하기 때문에 cross validation을 하여도 문제가 없습니다만,
전체 소스코드를 제가 보지 않아서 잘 모르겠습니다만, batch size를 현재보다 절반 정도 줄여 보시는 것은 어떨지요?&nbsp;

인프런 커뮤니티 질문&답변

image classication에서 cross validation 관련 질문