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
Interaktiv övning med praktiskt arbete
Gör teori till handling med en av våra interaktiva övningar
Starta övningen