• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

EPOCH이 돌지 않고 끝나는데 어떻게 하면 좋을까요?

22.12.20 12:24 작성 22.12.20 12:26 수정 조회수 195

0

선생님, 안녕하세요?

강의는 늘 잘 듣고 있습니다. BALLON데이터 셋 대신에 제가 학습해보고 싶은 데이터셋을 넣고 돌려보려고 합니다. 나머지는 잘 되었는데 마지막에서 epoch이 돌지 않고 끝나버립니다. 이럴 경우는 어떤 부분을 점검 하면 좋을까요?

(참고로 제 데이터는 gdrive에 넣고 마운트하였습니다. )

답변 1

답변을 작성해보세요.

0

안녕하십니까,

음, 올려주신것만 봐서는 원인을 잘 모르겠군요.

일단 balloon 데이터 세트로는 잘 학습이 되나요?

만약 그렇다면 dataset를 만드는게 잘못되었을 확률이 클 것 같습니다만,

cfg.data.train.ann_file = 'train_coco.json'

에서 train_coco.json 파일이 제대로 만들어졌는지 확인이 필요해 보입니다.

강의를 다시 한번 보시고 해당 파일이 coco 포맷으로 잘 만들어졌는지 확인 부탁드립니다.

안되시면 다시 글 부탁드립니다.