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

Cân trọng số loss

Ba phiên bản của mô hình hai đầu ra để dự đoán bảng chữ cái và ký tự mà bạn đã xây dựng trước đó đã được huấn luyện: model_a, model_bmodel_c. Với cả ba, loss được định nghĩa như sau:

loss_alpha = criterion(outputs_alpha, labels_alpha)
loss_char = criterion(outputs_char, labels_char)
loss = ((1 - char_weight) * loss_alpha) + (char_weight * loss_char)

Tuy nhiên, mỗi mô hình trong ba mô hình được huấn luyện với một char_weight khác nhau: 0.1, 0.5 hoặc 0.9.

Hãy dùng hàm bạn đã định nghĩa ở bài trước, evaluate_model(), để kiểm tra độ chính xác của từng mô hình. char_weight nào đã được dùng để huấn luyện từng mô hình?

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

Xem khóa học

Bài tập tương tác thực hành

Biến lý thuyết thành hành động với một trong các bài tập tương tác của chúng tôi

Bắt đầu bài tập