인프런 영문 브랜드 로고
인프런 영문 브랜드 로고

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

Hwaseong Nam님의 프로필 이미지
Hwaseong Nam

작성한 질문수

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

Oxford Pet 데이터 세트 Training 및 Object Detection 실습 02(Pipeline Config 수정및 Training 수행)

training 이 자동으로 중단되는 이유는 어떤 원인들이 있나요?

작성

·

271

0

안녕하세요.

tensorflow object detection api로 학습도중 step이 어느정도 진행되다가 자동으로 멈춰버리네요.

batch_size=1로 설정한 상태입니다.

이렇게 학습이 멈추는 이유가 보통 GPU 메모리가 부족해서 인가요?

답변 1

0

권 철민님의 프로필 이미지
권 철민
지식공유자

안녕하십니까,

학습이 멈추는 이유는 여러가지 이유가 있습니다. 그중 gpu 메모리 부족도 주요한 이유입니다. 하지만 batch_size=1 이라 메모리 부족으로 여겨지지는 않습니다. tensorflow object detection api는 버그가 많고 구현이 복잡해서 문제가 많습니다.

일단 오류 메시지를 여기에 올려주시겠습니까? 제가 확인해 보겠습니다.

Hwaseong Nam님의 프로필 이미지
Hwaseong Nam

작성한 질문수

질문하기