인프런 커뮤니티 질문&답변
작성한 질문수
EPOCH이 돌지 않고 끝나는데 어떻게 하면 좋을까요?
작성
·
362
·
수정됨
0
선생님, 안녕하세요?
강의는 늘 잘 듣고 있습니다. BALLON데이터 셋 대신에 제가 학습해보고 싶은 데이터셋을 넣고 돌려보려고 합니다. 나머지는 잘 되었는데 마지막에서 epoch이 돌지 않고 끝나버립니다. 이럴 경우는 어떤 부분을 점검 하면 좋을까요?
(참고로 제 데이터는 gdrive에 넣고 마운트하였습니다. )
답변 1
0
권 철민
지식공유자
안녕하십니까,
음, 올려주신것만 봐서는 원인을 잘 모르겠군요.
일단 balloon 데이터 세트로는 잘 학습이 되나요?
만약 그렇다면 dataset를 만드는게 잘못되었을 확률이 클 것 같습니다만,
cfg.data.train.ann_file = 'train_coco.json'
에서 train_coco.json 파일이 제대로 만들어졌는지 확인이 필요해 보입니다.
강의를 다시 한번 보시고 해당 파일이 coco 포맷으로 잘 만들어졌는지 확인 부탁드립니다.
안되시면 다시 글 부탁드립니다.