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

cvičení

Trénovací smyčka klasifikátoru obrázků

Čas natrénovat klasifikátor obrázků! Použiješ síť Net, kterou jsi definoval/a dříve, a naučíš ji rozlišovat mezi sedmi typy mraků.

Pro definici ztrátové funkce a optimalizátoru využiješ funkce z torch.nn a torch.optim, které jsou importované jako nn a optim. Samotnou trénovací smyčku měnit nemusíš – je stejná jako ty, které jsi psal/a dříve, jen s přidanou logikou pro výpis hodnoty ztráty během trénování.

Pokyny

100 XP
  • Definuj model pomocí své třídy Net s num_classes nastaveným na 7 a přiřaď ho do proměnné net.
  • Definuj ztrátovou funkci jako cross-entropy loss a přiřaď ji do criterion.
  • Definuj optimalizátor Adam, předej mu parametry modelu a rychlost učení 0.001, a přiřaď ho do optimizer.
  • Spusť trénovací for-cyklus iterací přes trénovací images a labels z dataloader_train.