Pondération des pertes
Trois versions du modèle à deux sorties pour la prédiction de l'alphabet et des caractères que vous avez construit précédemment ont été formées : model_a
, model_b
, et model_c
. Pour les trois, la perte a été définie comme suit :
loss_alpha = criterion(outputs_alpha, labels_alpha)
loss_char = criterion(outputs_char, labels_char)
loss = ((1 - char_weight) * loss_alpha) + (char_weight * loss_char)
Cependant, chacun des trois modèles a été entraîné avec un char_weight
différent : 0.1
, 0.5
ou 0.9
.
Utilisez la fonction que vous avez définie dans l'exercice précédent, evaluate_model()
, pour vérifier la précision de chaque modèle. Quel site char_weight
a été utilisé pour former quel modèle ?
Cet exercice fait partie du cours
Deep learning intermédiaire avec PyTorch
Exercice interactif pratique
Passez de la théorie à la pratique avec l’un de nos exercices interactifs
