1. Uczyć się
  2. /
  3. Courses
  4. /
  5. PyTorch Lightning으로 만드는 확장 가능한 AI 모델

Connected

Exercise

training step 구현하기

이 연습 문제에서는 이미지 분류 작업을 위한 PyTorch Lightning 모듈에서 training_step() 메서드를 구현해 보겠습니다. 배치에서 이미지와 레이블을 언패킹하고, forward 패스를 통해 모델 예측을 계산한 다음, cross entropy 손실을 구하고, 학습 손실을 로그로 남기세요.

Instrukcje

100 XP
  • forward 패스를 사용해 예측을 계산하세요.
  • cross entropy 손실을 계산하세요.
  • 학습 손실을 로그로 기록하세요.