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

연습 문제

학습과 평가

이 연습에서는 지금까지 배운 내용을 종합해, 실제 데이터셋인 에티오피아어 MNIST 손글씨 문자에 대해 신경망을 학습하고 평가해 보겠습니다.

ImageClassifier는 PyTorch Lightning으로 구현된 사전 정의 신경망 모델입니다. 특징 추출을 위한 합성곱 층, 비선형성을 도입하는 활성화 함수, 분류를 위한 완전연결 층으로 구성되어 있습니다.

Ethiopic MNIST 데이터셋은 미리 불러와 두었습니다.

지침

100 XP
  • Trainer를 가져오세요.
  • ImageClassifier 모델과 트레이너를 정의하세요.
  • 모델을 학습하세요.
  • 검증 세트에서 모델을 평가하세요.