Cấu hình optimizer
Giờ khi đã có logic huấn luyện, bạn cần chỉ định cách tối ưu các tham số của mô hình.
Trong bài tập này, bạn sẽ hoàn thiện phương thức configure_optimizers trong một mô-đun PyTorch Lightning dùng cho bài toán phân loại ảnh. Mục tiêu là thiết lập một optimizer để cập nhật tham số mô hình trong quá trình huấn luyện. Bạn sẽ dùng Adam với learning rate 1e-3 để hoàn thành yêu cầu này.
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ạo một Adam optimizer sử dụng các tham số của mô hình, đặt learning rate là
1e-3.
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
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer