작성
·
271
0
안녕하세요.
tensorflow object detection api로 학습도중 step이 어느정도 진행되다가 자동으로 멈춰버리네요.
batch_size=1로 설정한 상태입니다.
이렇게 학습이 멈추는 이유가 보통 GPU 메모리가 부족해서 인가요?
답변 1
0
안녕하십니까,
학습이 멈추는 이유는 여러가지 이유가 있습니다. 그중 gpu 메모리 부족도 주요한 이유입니다. 하지만 batch_size=1 이라 메모리 부족으로 여겨지지는 않습니다. tensorflow object detection api는 버그가 많고 구현이 복잡해서 문제가 많습니다.
일단 오류 메시지를 여기에 올려주시겠습니까? 제가 확인해 보겠습니다.