시작하기무료로 시작하기

평가 루프 작성하기

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

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

이 연습은 강의의 일부입니다

PyTorch로 배우는 딥러닝 입문

강의 보기

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

# Set the model to evaluation mode
____
validation_loss = 0.0

with torch.no_grad():
  
  for features, labels in validationloader:
    
      outputs = model(features)
      loss = criterion(outputs, labels)
      
      # Sum the current loss to the validation_loss variable
      validation_loss += ____
코드 편집 및 실행