Lightning Trainer 실행하기
이제 모델을 정의했으니, Lightning Trainer를 구성해 학습 루프를 실행해 보겠습니다. 이 연습 문제에서는 모델, 데이터 로더, 학습 설정을 어떻게 통합하는지 익힐 수 있어요. 이 구성을 제대로 이해하면 더 고급 학습으로 넘어가기 전에 큰 도움이 됩니다.
torch와 lightning.pytorch(별칭 pl)는 미리 로드되어 있습니다. 이전 연습에서 정의한 클래스를 사용해 초기화한 model도 준비되어 있어요.
이 연습은 강의의 일부입니다
PyTorch Lightning으로 만드는 확장 가능한 AI 모델
연습 안내
Trainer를 에폭 3으로 구성하세요.- 미리 정의된
model로 학습을 시작하세요.
실습형 인터랙티브 연습
이 예제를 이 샘플 코드를 완성하여 풀어보세요.
from lightning.pytorch import Trainer
# Configure the Trainer with max epochs
trainer = ____(max_epochs=____, devices=1)
# Start training with the predefined model
trainer.____(____, train_dataloader, val_dataloader)