train.py를 돌리기 위해 시도한 디버깅 목록 정리

데이터가 학습이 되지 않음

- 학습할 데이터의 차원을 모두 동일하게 셋팅

- 1600x1200x3으로 셋팅

- 셋팅하는 방법은 이전 게시물 참고할 것.

 

cuda 메모리 에러

- batch 수를 줄임 (하지만 batch 크기가 1이라서, 이를 해결하기 위해서는 gpu 교체가 필요함.)

- 학습할 이미지 개수/크기를 줄이는 것도 시도하였으나, 이건 메모리 에러에 큰 영향을 주지 않았음.

 

다음과 같이 학습이 된다.