Взвешивание функции потерь
Были обучены три версии модели с двумя выходами для предсказания алфавита и символа: model_a, model_b и model_c. Во всех трёх случаях функция потерь определялась следующим образом:
loss_alpha = criterion(outputs_alpha, labels_alpha)
loss_char = criterion(outputs_char, labels_char)
loss = ((1 - char_weight) * loss_alpha) + (char_weight * loss_char)
Однако каждая из трёх моделей обучалась с разным значением char_weight: 0.1, 0.5 или 0.9.
Воспользуйтесь функцией evaluate_model(), определённой в предыдущем упражнении, чтобы проверить точность каждой модели. Определите, с каким значением char_weight обучалась каждая модель.
Это упражнение является частью курса
Глубокое обучение на PyTorch: средний уровень
Практическое интерактивное упражнение
Превратите теорию в практику с помощью одного из наших интерактивных упражнений
Начать упражнение