Зважування функцій втрат
Було натреновано три версії двовихідної моделі для передбачення абетки та символа, яку ви будували раніше: 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 використали для тренування якої моделі?
Ця вправа є частиною курсу
Середній рівень Deep Learning з PyTorch
Практична інтерактивна вправа
Перетворіть теорію на практику за допомогою однієї з наших інтерактивних вправ
Почати вправу