Bắt đầu ngayBắt đầu miễn phí

Làm quen với LightningModule

Sẵn sàng xây dựng LightningModule đầu tiên của bạn! Trong bài tập thực hành này, bạn sẽ thiết lập cấu trúc cốt lõi của một quy trình phân loại. Bạn sẽ định nghĩa một tầng tuyến tính, truyền dữ liệu qua nó trong phương thức forward, và tính toán loss ở bước huấn luyện. Cấu trúc gọn gàng này tạo nền tảng vững chắc để bạn bắt đầu thử nghiệm với các mô hình của mình.

torch và lightning.pytorch (được import dưới tên pl) đã được tải sẵn cho bạ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

Xem khóa học

Hướng dẫn bài tập

  • Định nghĩa lớp LightModel kế thừa từ pl.LightningModule.
  • Định nghĩa một tầng tuyến tính để biến đổi đầu vào, giả sử có 16 đặc trưng đầu vào và 10 lớp đầu ra.

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.

# Define the model class
class LightModel(____):
  	# Define a linear layer to transform your input
    def __init__(self):
        super().__init__()
        self.layer = ____
    def forward(self, x):
        return self.layer(x)
    def training_step(self, batch, batch_idx):
        x, y = batch
        logits = self(x)
        loss = torch.nn.functional.cross_entropy(logits, y)
        return loss
Chỉnh sửa và Chạy Mã