Fonction de coût à utiliser avec callbacks_mixup().

🌐

Loss to be used with callbacks_mixup().

Pendant la phase d'entraînement, calcule la fonction de coût par rapport à deux cibles individuelles, leur alloue chacune un poids et les combine linéairement pour obtenir la perte moyenne du lot. Pour la validation et le test, se réfère au tenseur de perte passé.

Utilisation

nn_mixup_loss(loss)

Arguments

loss

Le fonction de coût sous-jacente de nn_module à appeler. Elle doit supporter le champ reduction. Pendant l'entraînement, l'attribut sera modifié en 'none' afin que nous obtenions le coût pour les observations individuelles. Voir par exemple la documentation du champ reduction dans torch::nn_cross_entropy_loss().

Détails

Doit être utilisé simultanément avec luz_callback_mixup().

Voir également

luz_callback_mixup()