1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Intermediate Deep Learning with PyTorch
Connected

cvičení

Trénování modelů s více výstupy

Při trénování modelů s více výstupy je zásadní správně definovat ztrátovou funkci.

V tomto případě model produkuje dva výstupy: predikce pro abecedu a pro znak. Ke každému z nich existují odpovídající pravdivostní popisky, které ti umožní vypočítat dvě samostatné ztráty: jednu za nesprávnou klasifikaci abecedy a druhou za nesprávnou klasifikaci znaku. Protože se v obou případech jedná o úlohu klasifikace více tříd, lze pokaždé použít Cross-Entropy loss.

Gradientový sestup však dokáže optimalizovat pouze jednu ztrátovou funkci. Celkovou ztrátu proto definuješ jako součet ztráty pro abecedu a ztráty pro znak.

Pokyny

100 XP
  • Vypočítej ztrátu klasifikace abecedy a přiřaď ji do proměnné loss_alpha.
  • Vypočítej ztrátu klasifikace znaku a přiřaď ji do proměnné loss_char.
  • Vypočítej celkovou ztrátu jako součet obou dílčích ztrát a přiřaď ji do proměnné loss.