ComeçarComece gratuitamente

Ponderação de perdas

Três versões do modelo de duas saídas para previsão de alfabeto e caracteres que você criou anteriormente foram treinadas: model_a, model_b, e model_c. Para todos os três, a perda foi definida da seguinte forma:

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

No entanto, cada um dos três modelos foi treinado com um char_weight diferente: 0.1, 0.5, ou 0.9.

Use a função que você definiu no exercício anterior, evaluate_model(), para verificar a precisão de cada modelo. Qual char_weight foi usado para treinar qual modelo?

Este exercício faz parte do curso

Aprendizagem profunda intermediária com PyTorch

Ver Curso

Exercício interativo prático

Transforme a teoria em ação com um de nossos exercícios interativos

Comece o exercício