Cài đặt bước huấn luyện
Trong bài tập này, bạn sẽ cài đặt phương thức training_step() trong một mô-đun PyTorch Lightning dùng cho bài toán phân loại ảnh.
Phần cài đặt của bạn cần tách batch thành ảnh và nhãn, tính dự đoán của mô hình qua forward pass, tính cross entropy loss, và ghi log loss huấn luyện.
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
- Đảm bảo bạn tính dự đoán bằng forward pass.
- Tính cross entropy loss.
- Ghi log loss huấn luyệ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 torch.nn.functional import cross_entropy
def training_step(self, batch, batch_idx):
x, y = batch
# Ensure that you compute predictions using the forward pass
y_hat = ____
# Calculate the cross entropy loss
loss = ____
# Log the loss
self.____("train_loss", loss)
return loss