1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch Lightning으로 만드는 확장 가능한 AI 모델
Connected

연습 문제

Torchmetrics로 모델 정확도 평가하기

모델이 얼마나 잘 동작하는지 평가하는 일은 매우 중요해요. 특히 배포를 준비할 때는 더더욱 그렇죠! Torchmetrics를 사용해 validation_step()에 정확도 계산을 자연스럽게 통합해 봅시다. 결과를 기록해 두면 모델의 향상을 쉽게 모니터링할 수 있으니, 로그 남기는 것도 잊지 마세요.

지침

100 XP
  • torchmetrics에서 Accuracy를 가져오세요.
  • __init__() 내부에서 정확도 metric을 인스턴스화하세요.
  • validation_step()에서 정확도를 계산하고 'val_acc'로 로그에 남기세요.