Kom igångKom igång gratis

Viktning av förlustfunktion

Tre versioner av modellen med två utdata för alfabets- och teckenprediktering som du byggde tidigare har tränats: model_a, model_b och model_c. För alla tre definierades förlusten enligt följande:

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

De tre modellerna tränades dock med olika värden på char_weight: 0.1, 0.5 eller 0.9.

Använd funktionen du definierade i föregående övning, evaluate_model(), för att kontrollera noggrannheten hos varje modell. Vilket char_weight användes för att träna vilken modell?

Den här övningen är en del av kursen

Fördjupad djupinlärning med PyTorch

Visa kurs

Interaktiv övning med praktiskt arbete

Gör teori till handling med en av våra interaktiva övningar

Starta övningen