image classication에서 cross validation 관련 질문

안녕하세요. 

강의 잘 듣고 있습니다. 머신러닝 완벽 가이드에서 CNN fundamental 편까지 잘 들었습니다.

강의 내용을 바탕으로 plant image classfication 에 cross validation을 적용해보려고, 

돌려보면 3번째 폴드를 어느 정도 돌고 나서는 OOM 이슈가 발생하는데 메모리 부족 현상을 해결할 수 있는 방법이 있을까요?

개인 워크스테이션을 사용하는데 cross validation을 적용하지 않으면 원활히 돌아가는데 KFold를 적용해 5 폴드 설정 후 

시행하면 OOM 이슈가 발생하네요.

좋은 답변 혹은 팁 부탁 드리겠습니다. 감사합니다.