Bộ tối ưu hóa
Đến lúc khám phá các bộ tối ưu hóa khác nhau mà bạn có thể dùng để huấn luyện mô hình.
Một hàm tùy chỉnh tên là train_model(optimizer, net, num_epochs) đã được định nghĩa sẵn cho bạn. Hàm này nhận bộ tối ưu hóa, mô hình và số epoch làm đầu vào, chạy các vòng lặp huấn luyện và in ra training loss ở cuối.
Hãy dùng train_model() để chạy vài lần huấn luyện ngắn với các bộ tối ưu hóa khác nhau và so sánh kết quả!
Bài tập này là một phần của khóa học
Deep Learning nâng cao với PyTorch
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.optim as optim
net = Net()
# Define the SGD optimizer
optimizer = optim.____(net.parameters(), lr=0.001)
train_model(
optimizer=optimizer,
net=net,
num_epochs=10,
)