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

Chungmin Myung님의 프로필 이미지
Chungmin Myung

작성한 질문수

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

batch_size관련질문

작성

·

109

0

선생님..

수업을 재밌게 듣고 있습니다.

Raccoon 데이터 세트 Training 및 Object Detection 실습 01에서 —steps=200 —batch_size=2 이 부분 설명해주실때, 좀 혼동이 가는 부분이 있어서 질문드립니다.

#질문 : --batch_size를 2로 변경하면은 steps를 변경해줘야 될것 같은데, 왜 변경 없이 200으로 입력하고 하는지 문의드립니다.

s = (n * e) / b

n = num of sample : 전체 학습할 데이터의 개수

e = epochs: Epoch 수

b = batch size: 배치 사이즈

s = steps: Step 수 

답변 1

0

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

안녕하십니까,

keras retinanet의 steps 는 tensorflow의 steps가 아니라 keras의 steps_per_epoch입니다.

때문에 s = n/b 입니다.  이미지 파일이 160개, batch_size가 2일 경우 80이 steps가 되지만, 이런 부연 설명을 굳이 드릴 필요 없을 거 같아서 200으로 통일했습니다.

감사합니다.

Chungmin Myung님의 프로필 이미지
Chungmin Myung

작성한 질문수

질문하기