Chạy Lightning Trainer
Giờ bạn đã định nghĩa xong mô hình, hãy cấu hình Trainer của Lightning để chạy vòng lặp huấn luyện. Bài tập này giúp bạn hiểu cách tích hợp mô hình, data loader và cấu hình huấn luyện — nắm vững thiết lập này là bước thiết yếu trước khi chuyển sang các kỹ thuật huấn luyện nâng cao.
torch và lightning.pytorch (được import dưới tên pl) đã được nạp sẵn. Một model cũng đã được khởi tạo, sử dụng lớp bạn định nghĩa ở bài trước.
Bài tập này là một phần của khóa học
Mô hình AI có khả năng mở rộng với PyTorch Lightning
Hướng dẫn bài tập
- Cấu hình
Trainervới 3 epoch. - Bắt đầu chạy huấn luyện với
modelđã được định nghĩa sẵn.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
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)