НачатьНачать бесплатно

Взвешивание функции потерь

Были обучены три версии модели с двумя выходами для предсказания алфавита и символа: 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: средний уровень

Посмотреть курс

Практическое интерактивное упражнение

Превратите теорию в практику с помощью одного из наших интерактивных упражнений

Начать упражнение