1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 배우는 딥러닝 입문
Connected

연습 문제

평가 루프 작성하기

이 연습 문제에서는 검증 손실을 계산하는 평가 루프를 작성해 보겠습니다. 평가 루프는 훈련 루프와 구조가 비슷하지만, 그래디언트 계산과 가중치 업데이트가 없습니다.

예측, 데이터 로딩, 손실 계산을 처리하기 위해 model, validationloader, 손실 함수 criterion이 이미 정의되어 있습니다.

지침 1/2

undefined XP
    1
    2
  • 모델을 평가 모드로 설정하세요.
  • 현재 배치의 손실을 validation_loss 변수에 합산하세요.