코드 질문

Question

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

# 평균 손실을 측정합니다

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가 계산되는 것이 아닌가요?

Answer

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

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

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

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

(1+1+1)/10 = 0.3

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

감사합니다.