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

작성자 없음

작성자 정보가 삭제된 글입니다.

TensorFlow 2.0으로 배우는 딥러닝 입문

TensorFlow 2.0과 ANN을 이용한 MNIST 숫자분류기 구현

코드 질문

작성

·

204

0

코드 중 하기 부분이 이해되지 않습니다.

# 평균 손실을 측정합니다

average_loss += current_loss/total_batch

 

batch for loop 안에서

average_loss += current_loss 를 계산하고

이후 epoch for loop 뒷단에서

average_loss = average_loss/total_batch

를 계산해야 1 epoch당 평균 loss가 계산되는 것이 아닌가요?

 

답변 1

0

AISchool님의 프로필 이미지
AISchool
지식공유자

안녕하세요~. 반갑습니다.

total_batch는 반복중 고정값이기 때문에 나눗셈을 먼저 수행해주냐 나중에 수행해주냐는 최종 계산결과에 차이가 없습니다.

예를 들어서, total_batch=10인 경우를 가정해보면

1/10 + 1/10 + 1/10 = 0.3

(1+1+1)/10 = 0.3

동일한 계산 결과값이 나오는 모습을 확인하실 수 있습니다.

감사합니다.

작성자 없음

작성자 정보가 삭제된 글입니다.

질문하기