Triển khai bước validation
Sau khi huấn luyện một mô hình mạng nơ-ron, bạn cần theo dõi hiệu năng của nó trong quá trình training. Với PyTorch Lightning, hãy triển khai phương thức validation_step() để tính và ghi log validation loss ở mỗi epoch.
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
- Tính dự đoán bằng mô hình trên batch đầu vào.
- Tính validation loss bằng
F.cross_entropy(). - Ghi log validation loss với
self.log()dưới tênval_loss.
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.
import torch.nn.functional as F
def validation_step(self, batch, batch_idx):
x, y = batch
# Compute predictions using the model
preds = ____(x)
# Calculate validation loss
loss = F.____(preds, y)
# Log the validation loss
self.____('val_loss', loss)