Pérdida de peso
Se han entrenado tres versiones del modelo de dos salidas para la predicción de alfabetos y caracteres que construiste anteriormente: model_a
, model_b
y model_c
. Para los tres, la pérdida se definió del siguiente modo:
loss_alpha = criterion(outputs_alpha, labels_alpha)
loss_char = criterion(outputs_char, labels_char)
loss = ((1 - char_weight) * loss_alpha) + (char_weight * loss_char)
Sin embargo, cada uno de los tres modelos se entrenó con un char_weight
diferente : 0.1
, 0.5
o 0.9
.
Utiliza la función que has definido en el ejercicio anterior, evaluate_model()
, para comprobar la precisión de cada modelo. ¿Qué char_weight
se utilizó para entrenar qué modelo?
Este ejercicio forma parte del curso
Aprendizaje profundo intermedio con PyTorch
Ejercicio interactivo práctico
Pon en práctica la teoría con uno de nuestros ejercicios interactivos
