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
Exercício interativo prático
Transforme a teoria em ação com um de nossos exercícios interativos
